fix: cast float to numpy array in function call (#39)

equinor · May 28, 2023 · 250928c · 250928c
1 parent 059dab5
commit 250928c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...ommon/libecalc/core/models/compressor/train/single_speed_compressor_train_common_shaft.py b/...ommon/libecalc/core/models/compressor/train/single_speed_compressor_train_common_shaft.py
@@ -947,7 +947,7 @@ def _calculate_train_result(mass_rate: float) -> CompressorTrainResultSingleTime
         # If solution not found along max speed curve, run at max_mass_rate, but using the defined pressure control.
         elif self.data_transfer_object.pressure_control is not None:
             if self._evaluate_rate_ps_pd(
-                rate=inlet_stream.mass_rate_to_standard_rate(mass_rate_kg_per_hour=max_mass_rate),
+                rate=np.asarray([inlet_stream.mass_rate_to_standard_rate(mass_rate_kg_per_hour=max_mass_rate)]),
                 suction_pressure=np.asarray([inlet_stream.pressure_bara]),
                 discharge_pressure=np.asarray([target_discharge_pressure]),
             )[0].is_valid: