Skip to content

Commit f1c75b8

Browse files
authored
fix: handle multistream compressor in turbine model check (#1253)
fixes: equinor/ecalc-internal#1288
1 parent b2630f2 commit f1c75b8

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

src/libecalc/domain/process/evaluation_input.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,13 @@ def periods(self) -> Periods:
5757
def apply_to_model(self, compressor_model: CompressorTrainModel | CompressorWithTurbineModel):
5858
rate_expr = self._rate_expression if isinstance(self._rate_expression, list) else [self._rate_expression]
5959

60-
if not isinstance(compressor_model, CompressorTrainCommonShaftMultipleStreamsAndPressures):
60+
model_to_check = (
61+
compressor_model.compressor_model
62+
if isinstance(compressor_model, CompressorWithTurbineModel)
63+
else compressor_model
64+
)
65+
66+
if not isinstance(model_to_check, CompressorTrainCommonShaftMultipleStreamsAndPressures):
6167
assert len(rate_expr) == 1
6268
stream_day_rate = np.asarray(rate_expr[0].get_stream_day_values(), dtype=np.float64)
6369
else:

0 commit comments

Comments
 (0)