fix: Replace 1-tuple enum values with ints (#22)

datamole-ai · Jul 26, 2023 · 2eab967 · 2eab967
1 parent f02bc88
commit 2eab967
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/edvart/data_types.py b/edvart/data_types.py
@@ -12,10 +12,10 @@
 class DataType(IntEnum):
     """Class describe possible data types."""
 
-    NUMERIC = (1,)
-    CATEGORICAL = (2,)
-    BOOLEAN = (3,)
-    DATE = (4,)
+    NUMERIC = 1
+    CATEGORICAL = 2
+    BOOLEAN = 3
+    DATE = 4
     UNKNOWN = 5
 
     def __str__(self):