small improvements

GetStream · Sep 18, 2023 · 76b4385 · 76b4385
1 parent 32317dc
commit 76b4385
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/openapi-gen/templates/python/type.tmpl b/openapi-gen/templates/python/type.tmpl
@@ -26,11 +26,13 @@
     {{- end}}
 {{- end}}
 
-{{- if (not .Schema.Enum)}}
+{{- if and (not .Schema.Enum) (not (gt (len .Schema.OneOf) 0))}}
 from dataclasses import dataclass, field
 from dataclasses_json import config, dataclass_json
-{{else}}
+{{else if .Schema.Enum}}
 from enum import Enum
+{{else if (gt (len .Schema.OneOf) 0)}}
+from dataclasses import dataclass
 {{- end}}
 
 {{- if .HasNonRequired}}