Fix coverage

koxudaxi · Jul 1, 2024 · 1675fb0 · 1675fb0
1 parent 46d8a7e
commit 1675fb0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/datamodel_code_generator/parser/base.py b/datamodel_code_generator/parser/base.py
@@ -1006,8 +1006,9 @@ def __collapse_root_models(
                             model_field.constraints = ConstraintsBase.merge_constraints(
                                 root_type_field.constraints, model_field.constraints
                             )
-                        if isinstance(  # no: pragma
-                            root_type_field, pydantic_model.DataModelField
+                        if isinstance(
+                            root_type_field,
+                            pydantic_model.DataModelField,  # no: pragma
                         ) and not model_field.extras.get('discriminator'):  # no: pragma
                             discriminator = root_type_field.extras.get('discriminator')
                             if discriminator:  # no: pragma