More

mobilityhouse · Jul 20, 2023 · 4fcf73a · 4fcf73a
1 parent 4938b26
commit 4fcf73a
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/scripts/v21/json_schema_parser.py b/scripts/v21/json_schema_parser.py
@@ -41,12 +41,11 @@ def _parse_as_object(name: str, parent: SubSchema, schema: Schema) -> Object:
             type = Number.from_sub_schema(sub_schema)
         elif is_array(sub_schema):
             if contains_reference(sub_schema["items"]):
-                _name = (
-                    sub_schema["items"]["$ref"]
-                    .split("/")[-1]
-                    .removesuffix("Type")
-                    .removesuffix("Enum")
-                )
+                _name = sub_schema["items"]["$ref"].split("/")[-1]
+                if _name.endswith("Enum"):
+                    _name = _name[:-4]
+                if _name.endswith("Type"):
+                    _name = _name[:-4]
 
                 sub_schema = resolve_reference(sub_schema["items"]["$ref"], schema)
                 if is_object(sub_schema):