Update test case

KevinMusgrave · Oct 11, 2023 · e4b1f39 · e4b1f39
1 parent 855baa6
commit e4b1f39
Showing 1 changed file with 1 addition and 15 deletions.
diff --git a/tests/losses/test_multilabel_supcon_loss.py b/tests/losses/test_multilabel_supcon_loss.py
@@ -90,6 +90,7 @@ def test_xbm_multisupcon_val(self):
                         self.test_xbm_multisupcon_val_gt[dtype][b], 
                         atol=1e-2 if dtype == torch.float16 else 1e-4))
 
+
     def test_with_no_valid_pairs(self):
         for dtype in TEST_DTYPES:
             embedding_angles = [0]
@@ -105,21 +106,6 @@ def test_with_no_valid_pairs(self):
             loss.backward()
             self.assertEqual(loss, 0)
 
-    def test_(self):
-        for dtype in TEST_DTYPES:
-            embedding_angles = [0]
-            embeddings = torch.tensor(
-                [angle_to_coord(a) for a in embedding_angles],
-                requires_grad=True,
-                dtype=dtype,
-            ).to(
-                TEST_DEVICE
-            )  # 2D embeddings
-            labels = torch.LongTensor([[0]])
-            loss = self.loss_func(embeddings, labels)
-            loss.backward()
-            self.assertEqual(loss, 0)
-
 
     def test_backward(self):
         for dtype in TEST_DTYPES: