Fix black error

HumanSignal · Aug 19, 2024 · 2feebbe · 2feebbe
1 parent 338522f
commit 2feebbe
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/adala/utils/pydantic_generator.py b/adala/utils/pydantic_generator.py
@@ -118,10 +118,7 @@ def json_schema_to_pydantic_type(
             else:
                 raise NotImplementedError(f"Unsupported JSON schema format: {format_}")
         elif "enum" in json_schema:
-            return Literal[*json_schema["enum"]]
-            # return Enum(
-            #     enum_class_name, {item: item for item in json_schema["enum"]}, type=str
-            # )
+            return Literal[tuple(json_schema["enum"])]
         return str
     elif type_ == "integer":
         return int