From bd650d1c94606bf46482dcb9a8bda451a67f223e Mon Sep 17 00:00:00 2001 From: Gard Spreemann Date: Wed, 16 Dec 2020 15:24:40 +0100 Subject: Disable test that requires PyTorch. --- debian/tests/upstream.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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))) -- cgit v1.2.3