chore: Sort index level 0 in pack_df_debug function and remove unnece…

…ssary code in _run.py
UMEP-dev · Aug 5, 2024 · 86367e2 · 86367e2
1 parent 9e5cd55
commit 86367e2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/src/supy/_post.py b/src/supy/_post.py
@@ -359,7 +359,7 @@ def pack_df_debug(dict_debug):
     )
     df_debug = (
         df_debug_raw.unstack(level=["group", "var"])
-        # .sort_index(level=0, axis=1)
+        .sort_index(level=0, axis=1)
         .dropna(axis=1, how="all")
     )
     return df_debug
diff --git a/src/supy/_run.py b/src/supy/_run.py
@@ -141,12 +141,8 @@ def suews_cal_tstep_multi(dict_state_start, df_forcing_block):
             "len_sim": np.array(df_forcing_block.shape[0], dtype=int),
         }
     )
-    # print("list_var_input_multitsteps", sorted(list_var_input_multitsteps))
-    # print("dict_input.keys()", sorted(dict_input.keys()))
-    # set_dif=set(dict_input.keys()) - set(list_var_input_multitsteps)
-    # print("set_dif", sorted(set_dif))
+
     dict_input = {k: dict_input[k] for k in list_var_input_multitsteps}
-    # dict_input = {k: dict_input[k] if k in dict_input else np.array([0.0]) for k in list_var_input_multitsteps}
 
     # main calculation:
 
@@ -434,7 +430,7 @@ def run_supy_ser(
 
         # collect debug info
         dict_debug = {
-            (tstep_final + freq, grid): debug
+            (tstep_final, grid): debug
             for grid, debug in zip(list_grid, list_dict_debug)
         }
         df_debug = pack_df_debug(dict_debug)