From f3aef79931d79a9bf313a40b3852c63d812c7dad Mon Sep 17 00:00:00 2001 From: andrewkho Date: Mon, 14 Oct 2024 15:14:06 -0700 Subject: [PATCH] fix import --- torchdata/nodes/map.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/torchdata/nodes/map.py b/torchdata/nodes/map.py index c4605d366..3a3078877 100644 --- 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