diff --git a/src/litdata/streaming/serializers.py b/src/litdata/streaming/serializers.py index 9e3b7665..62e386ed 100644 --- a/src/litdata/streaming/serializers.py +++ b/src/litdata/streaming/serializers.py @@ -405,7 +405,9 @@ 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'))}") + 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 index 952014ee..4d14b052 100644 --- a/tests/streaming/test_serializer.py +++ b/tests/streaming/test_serializer.py @@ -290,7 +290,6 @@ def test_tiff_serializer(): data, _ = serializer.serialize(file_path) assert isinstance(data, bytes) - # Deserialize deserialized_data = serializer.deserialize(data) assert isinstance(deserialized_data, np.ndarray)