Ensure filename remains set for later saving in sample_info (#2747)

Ensure filename remains set for later saving in sample_info --------- Co-authored-by: Nathan Voxland <[email protected]> Co-authored-by: FayazRahman <[email protected]>
activeloopai · Jan 24, 2024 · 15fba0c · 15fba0c
1 parent ac1c471
commit 15fba0c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/deeplake/core/chunk/sample_compressed_chunk.py b/deeplake/core/chunk/sample_compressed_chunk.py
@@ -53,8 +53,16 @@ def extend_if_has_space(self, incoming_samples: List[InputSample], update_tensor
                     num_samples += 1
                 else:
                     if serialized_sample:
+                        path = None
+                        if isinstance(incoming_sample, Sample):
+                            path = incoming_sample.path
+
                         sample = Sample(
-                            buffer=serialized_sample, compression=compr, shape=shape, dtype=dtype  # type: ignore
+                            buffer=serialized_sample,
+                            compression=compr,
+                            shape=shape,
+                            dtype=dtype,
+                            path=path,  # type: ignore
                         )
                         sample.htype = self.htype
                         incoming_samples[i] = sample

diff --git a/deeplake/core/sample.py b/deeplake/core/sample.py
@@ -19,6 +19,7 @@
     POINT_CLOUD_COMPRESSION,
     MESH_COMPRESSION,
     NIFTI_COMPRESSION,
+    BYTE_COMPRESSION,
 )
 from deeplake.util.exceptions import SampleReadError, UnableToReadFromUrlError
 from deeplake.util.exif import getexif
@@ -349,10 +350,7 @@ def _decompress(self, to_pil: bool = False):
                 )
 
         else:
-            if self.path and get_path_type(self.path) == "local":
-                compressed = self.path
-            else:
-                compressed = self.buffer
+            compressed = self.buffer
 
             if to_pil:
                 self._pil = decompress_array(