Skip to content

Commit 45d46fa

Browse files
committed
Simplify discriminator base mapping
1 parent 67a8dd5 commit 45d46fa

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

  • src/datamodel_code_generator/parser

src/datamodel_code_generator/parser/base.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1133,9 +1133,10 @@ def _get_discriminator_values(
11331133
if mapping:
11341134
_check_discriminator_mapping_paths(discriminator_model, mapping, discriminator_values)
11351135
if not discriminator_values:
1136-
for base_class in discriminator_model.base_classes:
1137-
if base_class.reference:
1138-
_check_discriminator_mapping_paths(base_class.reference, mapping, discriminator_values)
1136+
for base_reference in filter(
1137+
None, (base_class.reference for base_class in discriminator_model.base_classes)
1138+
):
1139+
_check_discriminator_mapping_paths(base_reference, mapping, discriminator_values)
11391140

11401141
return discriminator_values or [discriminator_model.path.split("/")[-1]]
11411142

0 commit comments

Comments
 (0)