diff options
author | tlacombe <lacombe1993@gmail.com> | 2020-07-20 11:41:13 +0200 |
---|---|---|
committer | tlacombe <lacombe1993@gmail.com> | 2020-07-20 11:41:13 +0200 |
commit | e94892f972357283e70c7534f84662dfaa21cc3e (patch) | |
tree | 5f5a69625f2c61201aaf9c31ecfeee95ad7b02ab /src/python/test/test_wasserstein_distance.py | |
parent | 107f8e6668509f5fd36e179f9a538b460d3941a9 (diff) |
update test enable_autodiff and _offdiag
Diffstat (limited to 'src/python/test/test_wasserstein_distance.py')
-rwxr-xr-x | src/python/test/test_wasserstein_distance.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/python/test/test_wasserstein_distance.py b/src/python/test/test_wasserstein_distance.py index 285b95c9..6701c7ba 100755 --- a/src/python/test/test_wasserstein_distance.py +++ b/src/python/test/test_wasserstein_distance.py @@ -31,7 +31,7 @@ def test_proj_on_diag(): def test_offdiag(): diag = np.array([[0, 1], [3, 5], [2, np.inf], [3, np.inf], [-np.inf, 8], [-np.inf, 12], [-np.inf, -np.inf], [np.inf, np.inf], [-np.inf, np.inf], [-np.inf, np.inf]]) - assert np.array_equal(_offdiag(diag), [[0, 1], [3, 5]]) + assert np.array_equal(_offdiag(diag, enable_autodiff=False), [[0, 1], [3, 5]]) def test_handle_essential_parts(): |