FIX six issue with module imports

andaag · larsmans · commit 25f72c4647c7 · 2014-08-01T10:49:29.000+02:00
Fixes #3518, #3519. Inspired by Johannes Schönberger's patch to scikit-image, scikit-image/scikit-image@71743d2
diff --git a/sklearn/utils/testing.py b/sklearn/utils/testing.py
@@ -608,7 +608,9 @@ def clean_warning_registry():
     """Safe way to reset warnings """
     warnings.resetwarnings()
     reg = "__warningregistry__"
-    for mod in sys.modules.copy().values():
+    for mod_name, mod in list(sys.modules.items()):
+        if 'six.moves' in mod_name:
+            continue
         if hasattr(mod, reg):
             getattr(mod, reg).clear()
 
