name refactoring

lightly-ai · Nov 19, 2024 · 8286ee3 · 8286ee3
1 parent f51e834
commit 8286ee3
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 10 deletions.
diff --git a/lightly/transforms/__init__.py b/lightly/transforms/__init__.py
@@ -59,7 +59,7 @@
 if torchvision_transforms_v2_available():
     from lightly.transforms.add_grid_transform import AddGridTransform
     from lightly.transforms.detcon_transform import (
-        DetConSimCLRViewTransform,
         DetConSTransform,
+        DetConSViewTransform,
     )
     from lightly.transforms.multi_view_transform_v2 import MultiViewTransformV2
diff --git a/lightly/transforms/detcon_transform.py b/lightly/transforms/detcon_transform.py
@@ -106,8 +106,8 @@ def __init__(
     ) -> None:
         self.grid_size = grid_size
 
-        tr1: List[Union[AddGridTransform, DetConSimCLRViewTransform]] = []
-        tr2: List[Union[AddGridTransform, DetConSimCLRViewTransform]] = []
+        tr1: List[Union[AddGridTransform, DetConSViewTransform]] = []
+        tr2: List[Union[AddGridTransform, DetConSViewTransform]] = []
 
         if self.grid_size is not None:
             grid_tr1 = AddGridTransform(
@@ -120,7 +120,7 @@ def __init__(
             tr2 += [grid_tr2]
 
         tr1 += [
-            DetConSimCLRViewTransform(
+            DetConSViewTransform(
                 gaussian_blur=gaussian_blur_t1,
                 input_size=input_size,
                 cj_prob=cj_prob,
@@ -141,7 +141,7 @@ def __init__(
             )
         ]
         tr2 += [
-            DetConSimCLRViewTransform(
+            DetConSViewTransform(
                 gaussian_blur=gaussian_blur_t2,
                 input_size=input_size,
                 cj_prob=cj_prob,

diff --git a/tests/transforms/test_detcon_transform.py b/tests/transforms/test_detcon_transform.py
@@ -7,10 +7,7 @@
     pytest.skip("torchvision.transforms.v2 not available", allow_module_level=True)
 from torchvision.tv_tensors import Image, Mask
 
-from lightly.transforms.detcon_transform import (
-    DetConSimCLRViewTransform,
-    DetConSTransform,
-)
+from lightly.transforms.detcon_transform import DetConSTransform, DetConSViewTransform
 
 
 # ignore typing due to Any type used in torchvison.transforms.v2.Transform
@@ -27,7 +24,7 @@ def mask() -> Mask:  # type: ignore[misc]
 
 class TestDetConSimCLRViewTransform:
     def test_given_masks(self, img: Image, mask: Mask) -> None:
-        tr = DetConSimCLRViewTransform(input_size=(224, 224))
+        tr = DetConSViewTransform(input_size=(224, 224))
 
         img_tr, mask_tr = tr(img, mask)
         assert img_tr.shape == (3, 224, 224)