remove second test

Lightning-AI · Nov 28, 2024 · 8c8dd78 · 8c8dd78
1 parent 8c0366d
commit 8c8dd78
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 12 deletions.
diff --git a/src/litdata/streaming/serializers.py b/src/litdata/streaming/serializers.py
@@ -405,9 +405,6 @@ def deserialize(self, data: bytes) -> Any:
         return tifffile.imread(io.BytesIO(data))  # This is a NumPy array
 
     def can_serialize(self, item: Any) -> bool:
-        print(
-            f"Checking can_serialize for: {item}, exists: {os.path.isfile(item)}, suffix: {item.lower().endswith(('.tif', '.tiff'))}"
-        )
         return isinstance(item, str) and os.path.isfile(item) and item.lower().endswith((".tif", ".tiff"))
 
 

diff --git a/tests/streaming/test_serializer.py b/tests/streaming/test_serializer.py
@@ -301,12 +301,3 @@ def test_tiff_serializer():
 
     # Clean up
     os.remove(file_path)
-
-
-def test_tiff_serializer_can_serialize():
-    serializer = TIFFSerializer()
-    assert serializer.can_serialize("image.tif")
-    assert serializer.can_serialize("image.tiff")
-    assert not serializer.can_serialize("image.jpg")
-    assert not serializer.can_serialize(None)
-    assert not serializer.can_serialize(123)