From 3682f9f7e36e1c3c4f4ef5b01344875f038e5087 Mon Sep 17 00:00:00 2001
From: cvergaras <154611098+cvergaras@users.noreply.github.com>
Date: Sun, 9 Jun 2024 09:42:57 +0000
Subject: [PATCH] Added restart_variables to init_profiles_dict

---
 glmpy/nml.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/glmpy/nml.py b/glmpy/nml.py
index dc8b041..ab5c87e 100644
--- a/glmpy/nml.py
+++ b/glmpy/nml.py
@@ -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
+            ) 
+            +
             "/"
         )
 
@@ -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
@@ -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, 
@@ -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(
@@ -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