Skip to content

Commit 438e106

Browse files
authored
hotfix active chunk range pytorch (#781)
1 parent 29ec1cd commit 438e106

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

hub/api/integrations.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -597,7 +597,8 @@ def _init_ds(self):
597597
if not self._inited:
598598
self._inited = True
599599
self._samples_in_chunks = {
600-
key: value.chunks[0] for key, value in self._ds._tensors.items()
600+
key: (None in value.shape) and 1 or value.chunks[0]
601+
for key, value in self._ds._tensors.items()
601602
}
602603
self._active_chunks = {}
603604
self._active_chunks_range = {}

0 commit comments

Comments
 (0)