small fix

activeloopai · Sep 5, 2023 · a554fff · a554fff
1 parent 4500621
commit a554fff
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 6 deletions.
diff --git a/deeplake/api/tests/test_info.py b/deeplake/api/tests/test_info.py
@@ -130,8 +130,6 @@ def test_update_reference_manually(local_ds_generator):
     l.append(99)
     ds.info.update()
 
-    ds = local_ds_generator()
-
     assert l == [1, 2, 3, 99]
 
 

diff --git a/deeplake/api/tests/test_meta.py b/deeplake/api/tests/test_meta.py
@@ -35,8 +35,6 @@ def test_subsequent_updates(local_ds_generator):
     assert len(ds) == 10
     assert ds.tensor.shape == (10, 100, 100)
 
-    ds = local_ds_generator()
-
     with local_ds_generator() as ds:
         for _ in range(5):
             ds.tensor.append(np.ones((100, 200)))

diff --git a/deeplake/enterprise/dataloader.py b/deeplake/enterprise/dataloader.py
@@ -717,6 +717,7 @@ def __get_indra_dataloader(
             ndim_dict=ndim_dict,
             tensor_info_dict=tensor_info_dict,
             offset=self._offset,
+            worker_init_fn=self.worker_init_fn,
         )
 
     def __iter__(self):
@@ -770,7 +771,6 @@ def __iter__(self):
                     htype_dict=htype_dict,
                     ndim_dict=ndim_dict,
                     tensor_info_dict=tensor_info_dict,
-                    worker_init_fn=self.worker_init_fn,
                 )
 
         dataset_read(self._orig_dataset)

diff --git a/deeplake/enterprise/test_pytorch.py b/deeplake/enterprise/test_pytorch.py
@@ -1,5 +1,4 @@
 import pickle
-from indra import api
 import deeplake
 import numpy as np
 import pytest