Merge pull request #35 from AllenNeuralDynamics/fix-allow-enum-object

Ensure non-native types implment `x-enumNames` attribute generation

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