Slight update to reading in forcing from yaml to allow a directory

Author: MatthewPaskin

src/supy/_load.py

@@ -815,11 +815,14 @@ def load_SUEWS_Forcing_met_df_yaml(path_forcing):
 
     if isinstance(path_forcing, (str, Path)):
         path_forcing = Path(path_forcing).resolve()
-        df_forcing_met = read_suews(path_forcing)
-    elif isinstance(path_forcing, list):
+        if path_forcing.is_dir():
+            path_forcing = sorted(path_forcing.glob('*'))
+        else:
+            df_forcing_met = read_suews(path_forcing)
+    if isinstance(path_forcing, list):
         path_forcing = [Path(p).resolve() for p in path_forcing]
         df_forcing_met = pd.concat([read_suews(fn) for fn in path_forcing])
-    else:
+    if not isinstance(path_forcing, (str, Path)) and not isinstance(path_forcing, list):
         import pdb; pdb.set_trace()
     # `drop_duplicates` in case some duplicates mixed
     df_forcing_met = df_forcing_met.drop_duplicates()

src/supy/_supy_module.py

@@ -237,7 +237,9 @@ def load_forcing_grid(
             df_forcing_met = load_SUEWS_Forcing_met_df_yaml(path_input)
             tstep_met_in = df_forcing_met.index[1] - df_forcing_met.index[0]
             tstep_met_in = int(tstep_met_in.total_seconds())
-            kdownzen = init_config_from_yaml(path=path_init).model.control.kdownzen.value
+            kdownzen = init_config_from_yaml(path=path_init).model.control.kdownzen
+            if kdownzen is not None:
+                kdownzen = kdownzen.value
             if kdownzen is None:
                 df_forcing_met_tstep = resample_forcing_met(
                     df_forcing_met, tstep_met_in, tstep_mod, lat, lon, alt, timezone