Fix DeprecationWarning for collections.Iterable import (#1417)

* Fix DeprecationWarning for collections.Iterable import * Simplify version comparison
pytorch · Oct 7, 2019 · a141363 · a141363
1 parent b13931a
commit a141363
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/torchvision/datasets/lsun.py b/torchvision/datasets/lsun.py
@@ -5,7 +5,11 @@
 import six
 import string
 import sys
-from collections import Iterable
+
+if sys.version_info < (3, 3):
+    from collections import Iterable
+else:
+    from collections.abc import Iterable
 
 if sys.version_info[0] == 2:
     import cPickle as pickle