Fix tests

lightly-ai · Sep 7, 2023 · 277746d · 277746d
1 parent a90a873
commit 277746d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/transforms/test_GaussianBlur.py → tests/transforms/test_gaussian_blur.py b/tests/transforms/test_GaussianBlur.py → tests/transforms/test_gaussian_blur.py
@@ -2,7 +2,7 @@
 
 from PIL import Image
 
-from lightly.transforms import GaussianBlur
+from lightly.transforms.gaussian_blur import GaussianBlur
 
 
 class TestGaussianBlur(unittest.TestCase):

diff --git a/tests/transforms/test_Jigsaw.py → tests/transforms/test_jigsaw.py b/tests/transforms/test_Jigsaw.py → tests/transforms/test_jigsaw.py
@@ -2,7 +2,7 @@
 
 from PIL import Image
 
-from lightly.transforms import Jigsaw
+from lightly.transforms.jigsaw import Jigsaw
 
 
 class TestJigsaw(unittest.TestCase):