Fix inline enum only string literal (#445)

Co-authored-by: Aaron Harper <aharper@duosecurity.com>
koxudaxi · Jun 9, 2021 · 03959a4 · 03959a4
1 parent bd02d06
commit 03959a4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/datamodel_code_generator/types.py b/datamodel_code_generator/types.py
@@ -16,6 +16,7 @@
     Tuple,
     Type,
     TypeVar,
+    Union,
 )
 
 from pydantic import create_model
@@ -71,7 +72,7 @@ class DataType(_BaseModel):
     is_dict: bool = False
     is_list: bool = False
     is_custom_type: bool = False
-    literals: List[str] = []
+    literals: List[Union[int, str]] = []
     use_standard_collections: bool = False
     use_generic_container: bool = False
     alias: Optional[str] = None