Skip to content

Commit 1e18322

Browse files
committed
Update torch datapipes
1 parent 99295d2 commit 1e18322

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tiledb/ml/readers/pytorch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
import sparse
99
import torchdata
1010
from torch.utils.data import DataLoader, IterDataPipe
11-
from torchdata.datapipes.iter import IterableWrapper
11+
from torch.utils.data.datapipes.iter import IterableWrapper
1212

1313
from ._pytorch_collators import Collator
1414
from ._tensor_schema import TensorSchema

0 commit comments

Comments
 (0)