diff --git a/src/DataSchemas/aind_behavior_services/utils.py b/src/DataSchemas/aind_behavior_services/utils.py index 6da8f50c..9a6f86d5 100644 --- a/src/DataSchemas/aind_behavior_services/utils.py +++ b/src/DataSchemas/aind_behavior_services/utils.py @@ -87,7 +87,7 @@ def enum_schema(self, schema: core_schema.EnumSchema) -> JsonSchemaValue: result["type"] = "array" _type = result.get("type", None) - if (self.render_x_enum_names) and (_type != "string") and (_type is not 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