fix: make sure default pressure control is set for all trains

Default pressure control was added to variable speed train, but forgotten for single speed and multiple streams.
equinor · Sep 26, 2024 · b233444 · b233444
1 parent 58c04a7
commit b233444
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/libecalc/presentation/yaml/mappers/model.py b/src/libecalc/presentation/yaml/mappers/model.py
@@ -85,7 +85,7 @@ def _pressure_control_mapper(
         YamlVariableSpeedCompressorTrainMultipleStreamsAndPressures,
     ],
 ) -> FixedSpeedPressureControl:
-    return FixedSpeedPressureControl(model_config.pressure_control.value)  # TODO: Could this be none previously?
+    return FixedSpeedPressureControl(model_config.pressure_control.value)
 
 
 def _get_curve_data_from_resource(resource: Resource, speed: float = 0.0):

diff --git a/src/libecalc/presentation/yaml/yaml_types/models/yaml_compressor_trains.py b/src/libecalc/presentation/yaml/yaml_types/models/yaml_compressor_trains.py
@@ -45,7 +45,7 @@ class YamlSingleSpeedCompressorTrain(YamlCompressorTrainBase):
         title="COMPRESSOR_TRAIN",
     )
     pressure_control: YamlPressureControl = Field(
-        None,
+        YamlPressureControl.DOWNSTREAM_CHOKE,
         description="Method for pressure control",
         title="PRESSURE_CONTROL",
     )
@@ -173,7 +173,7 @@ class YamlVariableSpeedCompressorTrainMultipleStreamsAndPressures(YamlCompressor
         title="STAGES",
     )
     pressure_control: YamlPressureControl = Field(
-        None,
+        YamlPressureControl.DOWNSTREAM_CHOKE,
         description="Method for pressure control",
         title="PRESSURE_CONTROL",
     )