FIX Check module is dict in clean_warning_registry

peay · peay · commit cdb1f802c074 · 2019-03-28T13:53:20.000+01:00
diff --git a/sklearn/utils/testing.py b/sklearn/utils/testing.py
@@ -780,7 +780,11 @@ def clean_warning_registry():
     reg = "__warningregistry__"
     for mod_name, mod in list(sys.modules.items()):
         if hasattr(mod, reg):
-            getattr(mod, reg).clear()
+            registry = getattr(mod, reg)
+            # `registry` may be `None` (e.g., if `py.apipkg` is used for
+            # module lazy loading).
+            if isinstance(registry, dict):
+                registry.clear()
 
 
 def check_skip_network():
