diff --git a/src/alfasim_score/converter/alfacase/convert_alfacase.py b/src/alfasim_score/converter/alfacase/convert_alfacase.py index 9348e99..f48af47 100644 --- 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 index 6c4434b..00da6ec 100644 --- 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"