fix: typing.Any is not imported for List[Any] (close #3)

Gowee · Jun 22, 2023 · e2d42bf · e2d42bf
1 parent b80e2ce
commit e2d42bf
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/target/python_class.rs b/src/target/python_class.rs
@@ -146,8 +146,11 @@ fn write_output(
                     );
                 }
             }
-            Type::Array(_) => {
+            Type::Array(inner) => {
                 imports_from_typing.insert("List");
+                if schema.arena.get(inner).unwrap().is_any() {
+                    imports_from_typing.insert("Any");
+                }
             }
             _ => {}
         }

diff --git a/src/target/python_inline.rs b/src/target/python_inline.rs
@@ -139,8 +139,11 @@ fn write_output(
                     });
                 }
             }
-            Type::Array(_) => {
+            Type::Array(inner) => {
                 imports_from_typing.insert("List");
+                if schema.arena.get(inner).unwrap().is_any() {
+                    imports_from_typing.insert("Any");
+                }
             }
             _ => {}
         }