bearstrong
diff --git a/‎lib/matplotlib/testing/decorators.py
Copy file name to clipboardExpand all lines: lib/matplotlib/testing/decorators.py
+1-1Lines changed: 1 addition & 1 deletion b/‎lib/matplotlib/testing/decorators.py
Copy file name to clipboardExpand all lines: lib/matplotlib/testing/decorators.py
+1-1Lines changed: 1 addition & 1 deletion
@@ -130,7 +130,7 @@ def cleanup(style=None):
     # writing a decorator with optional arguments.
 
     def make_cleanup(func):
-        if inspect.isgenerator(func):
+        if inspect.isgeneratorfunction(func):
             @functools.wraps(func)
             def wrapped_callable(*args, **kwargs):
                 original_units_registry = matplotlib.units.registry.copy()