summaryrefslogtreecommitdiff
path: root/src/python/test/test_dtm.py
diff options
context:
space:
mode:
authorMarc Glisse <marc.glisse@inria.fr>2020-04-19 12:30:35 +0200
committerMarc Glisse <marc.glisse@inria.fr>2020-04-19 12:30:35 +0200
commit8d9611206603f4f7506fe77a0273c73c9d67716b (patch)
tree20250b81f67ab830943cd0ba661fd7ddc1709a55 /src/python/test/test_dtm.py
parentacb9d5b9d1317d3d8168bc3ac46860d078abba84 (diff)
Drop redundant test
torch.isnan(None) raises an exception anyway
Diffstat (limited to 'src/python/test/test_dtm.py')
-rwxr-xr-xsrc/python/test/test_dtm.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/src/python/test/test_dtm.py b/src/python/test/test_dtm.py
index de74c42b..859189fa 100755
--- a/src/python/test/test_dtm.py
+++ b/src/python/test/test_dtm.py
@@ -46,14 +46,14 @@ def test_dtm_compare_euclidean():
r6 = dtm.fit_transform(pts2)
assert r6.detach().numpy() == pytest.approx(r0)
r6.sum().backward()
- assert pts2.grad is not None and not torch.isnan(pts2.grad).any()
+ assert not torch.isnan(pts2.grad).any()
pts2 = torch.tensor(pts, requires_grad=True)
assert pts2.grad is None
dtm = DistanceToMeasure(k, implementation="ckdtree", enable_autodiff=True)
r7 = dtm.fit_transform(pts2)
assert r7.detach().numpy() == pytest.approx(r0)
r7.sum().backward()
- assert pts2.grad is not None and not torch.isnan(pts2.grad).any()
+ assert not torch.isnan(pts2.grad).any()
def test_dtm_precomputed():