Merge pull request #124 from josephnowak/feature/general-improvements

The append and update method were refactored to improve its efficienc…
josephnowak · Sep 7, 2024 · 6e65a15 · 6e65a15
2 parents 4317c50 + 82ac6ff
commit 6e65a15
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tensordb/storages/zarr_storage.py b/tensordb/storages/zarr_storage.py
@@ -63,6 +63,8 @@ def __init__(
             synchronizer = synchronizer(path=lock_path)
 
         self.chunks = chunks
+        if self.chunks is None:
+            self.chunks = {}
         self.synchronizer = synchronizer
         self.unique_coords = unique_coords or {}
         self.sorted_coords = sorted_coords or {}
@@ -125,7 +127,7 @@ def _transform_to_dataset(self, new_data, chunk_data: bool = True) -> xr.Dataset
             new_data = new_data.to_dataset(name=self.data_names)
 
         if chunk_data:
-            new_data = new_data if self.chunks is None else new_data.chunk(self.chunks)
+            new_data = new_data.chunk(self.chunks)
         return new_data
 
     def append_preview(