conversion

activeloopai · May 10, 2023 · 7e02e05 · 7e02e05
1 parent 6bd93f3
commit 7e02e05
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/deeplake/tests/common.py b/deeplake/tests/common.py
@@ -142,9 +142,7 @@ def __exit__(self, *args, **kwargs):
 
 
 def convert_data_according_to_torch_version(batch):
-    import torch
-
-    if torch.__version__ < "2.0.0":
+    if isinstance(batch, List):
         return batch[0]
     else:
         return batch