Change the StrEnum to normal Enum;

PWPA-1933
ESSS · Jun 20, 2024 · 8f78504 · 8f78504
1 parent 99c53e5
commit 8f78504
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/alfasim_score/converter/alfacase/convert_alfacase.py b/src/alfasim_score/converter/alfacase/convert_alfacase.py
@@ -88,7 +88,7 @@ def _convert_casing_list(self) -> List[CasingSectionDescription]:
             for section in data["sections"]:
                 casing_sections.append(
                     CasingSectionDescription(
-                        name=f"{data['function']}_{data['type']}_{i}",
+                        name=f"{data['function'].value}_{data['type'].value}_{i}",
                         hanger_depth=section["top_md"],
                         settings_depth=section["base_md"],
                         hole_diameter=data["hole_diameter"],

diff --git a/src/alfasim_score/converter/alfacase/score_input_reader.py b/src/alfasim_score/converter/alfacase/score_input_reader.py
@@ -7,7 +7,7 @@
 import json
 from barril.units import Array
 from barril.units import Scalar
-from enum import StrEnum
+from enum import Enum
 from pathlib import Path
 
 from alfasim_score.constants import CEMENT_NAME
@@ -22,13 +22,13 @@
 from alfasim_score.units import YOUNG_MODULUS_UNIT
 
 
-class WellItemType(StrEnum):
+class WellItemType(str, Enum):
     DRILLING = "DRILLING"
     CASING = "CASING"
     NONE = "NONE"
 
 
-class WellItemFunction(StrEnum):
+class WellItemFunction(str, Enum):
     CONDUCTOR = "CONDUCTOR"
     SURFACE = "SURFACE"
     PRODUCTION = "PRODUCTION"