fixes memory leak (#1523)

activeloopai · Mar 5, 2022 · 364ce3a · 364ce3a
1 parent 6ef9381
commit 364ce3a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hub/core/chunk_engine.py b/hub/core/chunk_engine.py
@@ -396,7 +396,7 @@ def active_appended_chunk(self):
     @active_appended_chunk.setter
     def active_appended_chunk(self, value):
         if self.active_appended_chunk is not None:
-            self.cache.remove_hub_object(self.active_appended_chunk)
+            self.cache.remove_hub_object(self.active_appended_chunk.key)
         self._active_appended_chunk = value
         if value is not None:
             self.cache.register_hub_object(value.key, value)
@@ -408,7 +408,7 @@ def active_updated_chunk(self):
     @active_updated_chunk.setter
     def active_updated_chunk(self, value):
         if self.active_updated_chunk is not None:
-            self.cache.remove_hub_object(self.active_updated_chunk)
+            self.cache.remove_hub_object(self.active_updated_chunk.key)
         self._active_updated_chunk = value
         if value is not None:
             self.cache.register_hub_object(value.key, value)