Fixed iterator/iterable support for generators in TypeChecker

Fixes #77.
agronholm · Sep 15, 2019 · c684a1c · c684a1c
1 parent d62db01
commit c684a1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/typeguard/__init__.py b/typeguard/__init__.py
@@ -901,7 +901,7 @@ def __call__(self, frame, event: str, arg) -> None:  # pragma: no cover
                     return_type_hint = memo.type_hints['return']
                     if return_type_hint is not None:
                         origin = getattr(return_type_hint, '__origin__', None)
-                        if origin in (collections.abc.Generator, Generator):
+                        if origin in generator_origin_types:
                             # Check the types of the yielded values
                             memo.type_hints['return'] = return_type_hint.__args__[0]
                 else: