Simplify tigramite helper function

climateintelligence · Mar 25, 2024 · 7c21f22 · 7c21f22
1 parent ac80116
commit 7c21f22
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/hawk/analysis/pcmci_tools.py b/hawk/analysis/pcmci_tools.py
@@ -46,8 +46,10 @@ def initialize_tigramite_df(df: pd.DataFrame):
     :return: tigramite dataframe and variable names tuple
     """
 
-    var_names = df.columns
+    dataframe = pp.DataFrame(
+        df.values,
+        datatime={0: np.arange(len(df))},
+        var_names=df.columns,
+    )
 
-    dataframe = pp.DataFrame(df.values, datatime={0: np.arange(len(df))}, var_names=var_names)
-
-    return dataframe, var_names
+    return dataframe