formatted code

Author: KevinMusgrave

src/pytorch_metric_learning/datasets/__init__.py

@@ -2,4 +2,4 @@
 from .cars196 import Cars196
 from .cub import CUB
 from .inaturalist2018 import INaturalist2018
-from .sop import StanfordOnlineProducts
+from .sop import StanfordOnlineProducts

src/pytorch_metric_learning/losses/generic_pair_loss.py

@@ -39,7 +39,9 @@ def pair_based_loss(self, mat, indices_tuple):
         if len(a2) > 0:
             neg_pair = mat[a2, n]
         return self._compute_loss(pos_pair, neg_pair, indices_tuple)
-    
+
     @staticmethod
     def _assert_either_pos_or_neg(pos_mask, neg_mask):
-        assert not torch.any((pos_mask != 0) & (neg_mask != 0)), "Each pair should be either be positive or negative"
+        assert not torch.any(
+            (pos_mask != 0) & (neg_mask != 0)
+        ), "Each pair should be either be positive or negative"

src/pytorch_metric_learning/losses/subcenter_arcface_loss.py

@@ -19,7 +19,11 @@ def __init__(self, *args, margin=28.6, scale=64, sub_centers=3, **kwargs):
         del kwargs["num_classes"]
         del kwargs["embedding_size"]
         super().__init__(
-            num_classes=num_classes * sub_centers, embedding_size=embedding_size, margin=margin, scale=scale, **kwargs
+            num_classes=num_classes * sub_centers,
+            embedding_size=embedding_size,
+            margin=margin,
+            scale=scale,
+            **kwargs
         )
         self.sub_centers = sub_centers
         self.num_classes = num_classes

tests/datasets/test_cars196.py

@@ -6,6 +6,7 @@
 from torch.utils.data import DataLoader
 
 from pytorch_metric_learning.datasets import Cars196
+
 from .. import TEST_DATASETS
 
 

tests/datasets/test_cub.py

@@ -6,6 +6,7 @@
 from torch.utils.data import DataLoader
 
 from pytorch_metric_learning.datasets import CUB
+
 from .. import TEST_DATASETS
 
 

tests/datasets/test_inaturalist2018.py

@@ -6,6 +6,7 @@
 from torch.utils.data import DataLoader
 
 from pytorch_metric_learning.datasets import INaturalist2018
+
 from .. import TEST_DATASETS
 
 

tests/datasets/test_sop.py

@@ -6,6 +6,7 @@
 from torch.utils.data import DataLoader
 
 from pytorch_metric_learning.datasets import StanfordOnlineProducts
+
 from .. import TEST_DATASETS
 
 

tests/losses/test_subcenter_arcface_loss.py

@@ -152,10 +152,9 @@ def test_reducer_subcenter_arcface(self):
             embedding_size=64,
             reducer=DoNothingReducer(),
         )
-        
+
         emb = torch.randn(4, 64)
         result = arcfaceloss(emb, torch.arange(4))
 
         self.assertTrue(isinstance(result, dict))
-        self.assertTrue(result['loss']['losses'].shape[0] == 4)
-        
+        self.assertTrue(result["loss"]["losses"].shape[0] == 4)