Use StrEnum in python 3.11

equinor · Nov 30, 2023 · 4f8a144 · 4f8a144
1 parent 10c7280
commit 4f8a144
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/webviz_ert/models/data_model.py b/webviz_ert/models/data_model.py
@@ -1,12 +1,21 @@
-from enum import Enum
+import sys
 
+if sys.version_info < (3, 11):
+    from enum import Enum
 
-class DataType(str, Enum):
+    class StrEnum(str, Enum):
+        pass
+
+else:
+    from enum import StrEnum
+
+
+class DataType(StrEnum):
     RESPONSE = "resp"
     PARAMETER = "param"
     ENSEMBLE = "ens"
 
 
-class AxisType(str, Enum):
+class AxisType(StrEnum):
     INDEX = "index"
     TIMESTAMP = "timestamp"