diff options
| -rw-r--r-- | debian/tests/upstream.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/debian/tests/upstream.py b/debian/tests/upstream.py index 1b4caffe..3c23027c 100644 --- a/debian/tests/upstream.py +++ b/debian/tests/upstream.py @@ -6,7 +6,8 @@ blacklist = ["test_dtm.py", # Requires hnswlib "test_knn.py", # Requires pykeops "test_representations.py", # Look into this "test_tomato.py", # Wants to use X. - "test_wasserstein_distance.py::test_wasserstein_distance_grad" # Requires PyTorch. + "test_wasserstein_distance.py::test_wasserstein_distance_grad", # Requires PyTorch. + "test_wasserstein_with_tensors.py" # Requires PyTorch and TF. ] print("Blacklist: %s" %(str(blacklist))) |