Merge pull request #2300 from activeloopai/fr_local_org_id

Set org_id for local datasets if environment variable is set
activeloopai · Apr 21, 2023 · 450cd72 · 450cd72
2 parents b1c0e28 + e5d371e
commit 450cd72
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/deeplake/core/dataset/dataset.py b/deeplake/core/dataset/dataset.py
@@ -39,6 +39,7 @@
     MB,
     SAMPLE_INFO_TENSOR_MAX_CHUNK_SIZE,
     DEFAULT_READONLY,
+    ENV_HUB_DEV_USERNAME,
 )
 from deeplake.core.fast_forwarding import ffw_dataset_meta
 from deeplake.core.index import Index
@@ -1655,7 +1656,8 @@ def _populate_meta(self, verbose=True):
             self.flush()
 
     def _register_dataset(self):
-        """overridden in DeepLakeCloudDataset"""
+        if not self.__dict__["org_id"]:
+            self.org_id = os.environ.get(ENV_HUB_DEV_USERNAME)
 
     def _send_query_progress(self, *args, **kwargs):
         """overridden in DeepLakeCloudDataset"""