[python] Fix postponed model imports (#16882)

* replaced method used on model to import the model not the mapping name * replaced method used on model to import the model not the mapping name
OpenAPITools · Nov 1, 2023 · 4402635 · 4402635
1 parent 4302d37
commit 4402635
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/...api-generator/src/main/java/org/openapitools/codegen/languages/AbstractPythonCodegen.java b/...api-generator/src/main/java/org/openapitools/codegen/languages/AbstractPythonCodegen.java
@@ -929,7 +929,7 @@ private ModelsMap postProcessModelsMap(ModelsMap objs) {
                     typingImports.add("Union");
                     Set<CodegenDiscriminator.MappedModel> discriminator = model.getDiscriminator().getMappedModels();
                     for (CodegenDiscriminator.MappedModel mappedModel : discriminator) {
-                        postponedModelImports.add(mappedModel.getMappingName());
+                        postponedModelImports.add(mappedModel.getModelName());
                     }
                 }
             }

diff --git a/...tor/src/main/java/org/openapitools/codegen/languages/AbstractPythonPydanticV1Codegen.java b/...tor/src/main/java/org/openapitools/codegen/languages/AbstractPythonPydanticV1Codegen.java
@@ -881,7 +881,7 @@ private ModelsMap postProcessModelsMap(ModelsMap objs) {
                     typingImports.add("Union");
                     Set<CodegenDiscriminator.MappedModel> discriminator = model.getDiscriminator().getMappedModels();
                     for (CodegenDiscriminator.MappedModel mappedModel : discriminator) {
-                        postponedModelImports.add(mappedModel.getMappingName());
+                        postponedModelImports.add(mappedModel.getModelName());
                     }
                 }
             }