Skip to content

Commit

Permalink
Added restart_variables to init_profiles_dict
Browse files Browse the repository at this point in the history
  • Loading branch information
cvergaras committed Jun 9, 2024
1 parent a5d79bd commit 3682f9f
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion glmpy/nml.py
Original file line number Diff line number Diff line change
Expand Up @@ -508,6 +508,10 @@ def _write_nml_init_profiles(self, init_profiles: dict) -> str:
self.nml_param_val(
init_profiles, "wq_init_vals", self.nml_list
) +
self.nml_param_val(
init_profiles, "restart_variables", self.nml_list
)
+
"/"
)

Expand Down Expand Up @@ -1746,6 +1750,7 @@ def __init__(
num_wq_vars: Union[int, None] = None,
wq_names: Union[List[str], str, None] = None,
wq_init_vals: Union[List[float], float, None] = None,
restart_variables: Union[List[float], float, None] = None,
):
self.lake_depth = lake_depth
self.num_depths = num_depths
Expand All @@ -1755,6 +1760,7 @@ def __init__(
self.num_wq_vars = num_wq_vars
self.wq_names = wq_names
self.wq_init_vals = wq_init_vals
self.restart_variables = restart_variables

def __call__(
self,
Expand Down Expand Up @@ -1806,6 +1812,7 @@ def __call__(
self.the_depths = self._single_value_to_list(self.the_depths)
self.wq_names = self._single_value_to_list(self.wq_names)
self.wq_init_vals = self._single_value_to_list(self.wq_init_vals)
self.restart_variables = self._single_value_to_list(self.restart_variables)

if check_errors:
warnings.warn(
Expand All @@ -1823,7 +1830,8 @@ def __call__(
"the_sals": self.the_sals,
"num_wq_vars": self.num_wq_vars,
"wq_names": self.wq_names,
"wq_init_vals": self.wq_init_vals
"wq_init_vals": self.wq_init_vals,
"restart_variables": self.restart_variables
}

return init_profiles_dict
Expand Down

0 comments on commit 3682f9f

Please sign in to comment.