Fix AttributeError: __args__ on Python 3.9.

Instagram · Jun 5, 2021 · a5d5ee3 · a5d5ee3
1 parent 9c81337
commit a5d5ee3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -4,6 +4,9 @@ Changelog
 main
 ----
 
+* Fix `AttributeError: __args__` when generating stubs on Python 3.9. Thanks
+  GameDungeon and ntjess for the report. Fixes #231.
+
 
 21.5.0
 ------

diff --git a/monkeytype/stubs.py b/monkeytype/stubs.py
@@ -140,7 +140,7 @@ def get_imports_for_annotation(anno: Any) -> ImportMap:
         else:
             imports[anno.__module__].add(
                 _get_import_for_qualname(qualname_of_generic(anno)))
-        elem_types = anno.__args__ or []
+        elem_types = getattr(anno, "__args__", None) or []
         for et in elem_types:
             elem_imports = get_imports_for_annotation(et)
             imports.merge(elem_imports)