Skip to content

Commit 476e3a8

Browse files
committed
fix: adding conversion factor argument to ensemble as well
1 parent 95def8a commit 476e3a8

1 file changed

Lines changed: 6 additions & 3 deletions

File tree

rocketpy/environment/environment.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1338,7 +1338,7 @@ def set_atmospheric_model( # pylint: disable=too-many-statements
13381338
dataset, dictionary, conversion_factor=conversion_factor
13391339
)
13401340
else:
1341-
self.process_ensemble(dataset, dictionary)
1341+
self.process_ensemble(dataset, dictionary, conversion_factor)
13421342
case _: # pragma: no cover
13431343
raise ValueError(f"Unknown model type '{type}'.")
13441344

@@ -2025,7 +2025,7 @@ def process_forecast_reanalysis(self, file, dictionary, conversion_factor): # p
20252025
# Close weather data
20262026
data.close()
20272027

2028-
def process_ensemble(self, file, dictionary): # pylint: disable=too-many-locals,too-many-statements
2028+
def process_ensemble(self, file, dictionary, conversion_factor): # pylint: disable=too-many-locals,too-many-statements
20292029
"""Import and process atmospheric data from weather ensembles
20302030
given as ``netCDF`` or ``OPeNDAP`` files. Sets pressure, temperature,
20312031
wind-u and wind-v profiles and surface elevation obtained from a weather
@@ -2076,6 +2076,9 @@ def process_ensemble(self, file, dictionary): # pylint: disable=too-many-locals
20762076
"u_wind": "ugrdprs",
20772077
"v_wind": "vgrdprs",
20782078
}
2079+
conversion_factor : float, int
2080+
Specifies the factor by which the pressure will be multiplied
2081+
in order to transform it to Pascal.
20792082
20802083
See also
20812084
--------
@@ -2140,7 +2143,7 @@ class for some dictionary examples.
21402143
num_members = 1
21412144

21422145
# Get pressure level data from file
2143-
levels = get_pressure_levels_from_file(data, dictionary)
2146+
levels = get_pressure_levels_from_file(data, dictionary, conversion_factor)
21442147

21452148
inverse_dictionary = {v: k for k, v in dictionary.items()}
21462149
param_dictionary = {

0 commit comments

Comments
 (0)