[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
Lightning-AI · Nov 28, 2024 · 8c0366d · 8c0366d
1 parent ce08419
commit 8c0366d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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)