Use try/except for python versions

equinor · Nov 30, 2023 · 8d58d02 · 8d58d02
1 parent 4f8a144
commit 8d58d02
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/webviz_ert/models/data_model.py b/webviz_ert/models/data_model.py
@@ -1,13 +1,12 @@
-import sys
+try:
+    from enum import StrEnum
 
-if sys.version_info < (3, 11):
+except ImportError:
     from enum import Enum
 
     class StrEnum(str, Enum):
-        pass
-
-else:
-    from enum import StrEnum
+        def __str__(self):
+            return self.value
 
 
 class DataType(StrEnum):