diff --git a/src/DataSchemas/aind_behavior_services/utils.py b/src/DataSchemas/aind_behavior_services/utils.py index 33b8d93e..9a6f86d5 100644 --- a/src/DataSchemas/aind_behavior_services/utils.py +++ b/src/DataSchemas/aind_behavior_services/utils.py @@ -24,7 +24,7 @@ def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) self.nullable_as_oneof = kwargs.get("nullable_as_oneof", True) self.unions_as_oneof = kwargs.get("unions_as_oneof", True) - self.render_xenum_names = kwargs.get("render_xenum_names", True) + self.render_x_enum_names = kwargs.get("render_x_enum_names", True) def nullable_schema(self, schema: core_schema.NullableSchema) -> JsonSchemaValue: null_schema = {"type": "null"} @@ -86,7 +86,8 @@ def enum_schema(self, schema: core_schema.EnumSchema) -> JsonSchemaValue: elif types == {list}: result["type"] = "array" - if (self.render_xenum_names) and (result["type"] != "string"): + _type = result.get("type", None) + if (self.render_x_enum_names) and (_type != "string"): result["x-enumNames"] = [screaming_snake_case_to_pascal_case(v.name) for v in schema["members"]] return result