corrected tensor calculation logic

activeloopai · May 10, 2023 · 6bd93f3 · 6bd93f3
1 parent 90d0614
commit 6bd93f3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/deeplake/integrations/pytorch/shuffle_buffer.py b/deeplake/integrations/pytorch/shuffle_buffer.py
@@ -118,6 +118,8 @@ def _num_torch_tensors(self, sample):
             return 0
         if isinstance(sample, TorchTensor):
             return 1
+        elif isinstance(sample, bytes):
+            return 0
         elif isinstance(sample, dict):
             return sum(self._num_torch_tensors(tensor) for tensor in sample.values())
         elif isinstance(sample, Sequence):