Merge pull request #18 from voletiv/master

Fix bug : convert to cpu() before np assert
fbcotter · Oct 21, 2020 · 9d7018b · 9d7018b
2 parents 5296411 + 0b6312e
commit 9d7018b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_rowfilter.py b/tests/test_rowfilter.py
@@ -113,7 +113,7 @@ def test_equal_numpy_qshift2():
     im_t = torch.unsqueeze(torch.tensor(im, dtype=torch.float32), dim=0).to(dev)
     ref = ref_rowfilter(im, h)
     y = rowfilter(im_t, prep_filt(h, 1).to(dev))
-    np.testing.assert_array_almost_equal(y[0], ref, decimal=4)
+    np.testing.assert_array_almost_equal(y[0].cpu(), ref, decimal=4)
 
 
 @pytest.mark.skip