From 2929678305e4d0aa4640d72b0df009ed291d57ee Mon Sep 17 00:00:00 2001
From: "Dr. Kashif Rasul" <kashif.rasul@gmail.com>
Date: Fri, 18 Dec 2020 13:06:51 +0100
Subject: [PATCH] fix import

---
 pts/dataset/loader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pts/dataset/loader.py b/pts/dataset/loader.py
index 4a21409..d74cedd 100644
--- a/pts/dataset/loader.py
+++ b/pts/dataset/loader.py
@@ -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