test actually works (kornia#2510)

johnnv1 · Aug 11, 2023 · 796045f · 796045f
1 parent 3c747cc
commit 796045f
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/test/feature/test_matching.py b/test/feature/test_matching.py
@@ -468,11 +468,8 @@ def test_real(self, device, dtype, data):
         assert idxs.shape[0] == dists.shape[0]
         assert dists.shape[0] <= data_dev['descs1'].shape[0]
         assert dists.shape[0] <= data_dev['descs2'].shape[0]
-        # key doesn't exist in data_dev
-        # expected_idxs = data_dev['lightglue_disk_idxs'].long()
-        data_dev['expected_idxs'].long()
-        # TODO: fix this
-        # assert_close(idxs, expected_idxs, rtol=1e-4, atol=1e-4)
+        expected_idxs = data_dev['lightglue_disk_idxs'].long()
+        assert_close(idxs, expected_idxs, rtol=1e-4, atol=1e-4)
 
     @pytest.mark.parametrize("data", ["lightglue_idxs"], indirect=True)
     def test_single_nocrash(self, device, dtype, data):