fix import

This commit is contained in:
Dr. Kashif Rasul
2020-12-18 13:06:51 +01:00
parent 2726bc94ec
commit 2929678305
+1 -1
View File
@@ -1,5 +1,5 @@
from typing import Callable, Iterable, Iterator, List, Optional
from torch.utils import data
import itertools
from torch.utils.data import IterableDataset