Fix DeprecationWarning for collections.Iterable import (#1417)

* Fix DeprecationWarning for collections.Iterable import * Simplify version comparison

Fix DeprecationWarning for collections.Iterable import (#1417)
* Fix DeprecationWarning for collections.Iterable import * Simplify version comparison
a141363f · Adam J. Stewart · Francisco Massa · b13931af · a141363f
Commit a141363f authored Oct 07, 2019 by Adam J. Stewart Committed by Francisco Massa Oct 07, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

torchvision/datasets/lsun.py torchvision/datasets/lsun.py +5 -1

No files found.
--- a/torchvision/datasets/lsun.py
+++ b/torchvision/datasets/lsun.py
@@ -5,7 +5,11 @@ import os.path
 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