Fixing the deepmemory error

activeloopai · Dec 28, 2023 · d20e4d6 · d20e4d6
1 parent 1604447
commit d20e4d6
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 8 deletions.
diff --git a/deeplake/core/vectorstore/dataset_handlers/embedded_dataset_handler.py b/deeplake/core/vectorstore/dataset_handlers/embedded_dataset_handler.py
@@ -13,6 +13,7 @@
 from deeplake.core.vectorstore import utils
 from deeplake.core.vectorstore.dataset_handlers.dataset_handler_base import DHBase
 from deeplake.core.vectorstore.deep_memory.deep_memory import (
+    DeepMemory,
     use_deep_memory,
 )
 from deeplake.core.vectorstore.vector_search import dataset as dataset_utils
@@ -84,6 +85,15 @@ def __init__(
         self.tensor_params = tensor_params
         self.distance_metric_index = index_maintenance.index_operation_vectorstore(self)
 
+        self.deep_memory = DeepMemory(
+            dataset=self.dataset,
+            path=self.path,
+            token=self.token,
+            logger=self.logger,
+            embedding_function=self.embedding_function,
+            creds=self.creds,
+        )
+
     def add(
         self,
         embedding_function: Union[Callable, List[Callable]],

diff --git a/deeplake/core/vectorstore/dataset_handlers/managed_dataset_handler.py b/deeplake/core/vectorstore/dataset_handlers/managed_dataset_handler.py
@@ -129,6 +129,7 @@ def __init__(
             logger.info(response.summary)
 
         self.deep_memory = DeepMemory(
+            dataset=None,
             path=self.path,
             token=self.token,
             logger=self.logger,

diff --git a/deeplake/core/vectorstore/deeplake_vectorstore.py b/deeplake/core/vectorstore/deeplake_vectorstore.py
@@ -138,14 +138,7 @@ def __init__(
             **kwargs,
         )
 
-        self.deep_memory = DeepMemory(
-            dataset=self.dataset_handler.dataset,
-            path=self.dataset_handler.path,
-            token=self.dataset_handler.token,
-            logger=logger,
-            embedding_function=embedding_function,
-            creds=self.dataset_handler.creds,
-        )
+        self.deep_memory = self.dataset_handler.deep_memory
 
     def add(
         self,