fix import

pytorch · Oct 15, 2024 · f3aef79 · f3aef79
1 parent b377e9b
commit f3aef79
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/torchdata/nodes/map.py b/torchdata/nodes/map.py
@@ -2,12 +2,11 @@
 # import multiprocessing as mp
 import queue
 import threading
-from typing import Callable, List, Literal, TypeVar, Union
+from typing import Callable, Iterator, List, Literal, TypeVar, Union
 
 import torch.multiprocessing as mp
 
 from torch._utils import ExceptionWrapper
-from torch.utils.data.sampler import Iterator
 
 from torchdata.nodes import BaseNode, T