fix bug

RL4H · Dec 3, 2024 · 26e079b · 26e079b
1 parent 534c25b
commit 26e079b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/visualiser/notebooks/explain_DT.py b/visualiser/notebooks/explain_DT.py
@@ -66,7 +66,7 @@ def get_corr(cohort, sub):
 
     return df_combined
 
-def decompose_data(df, columns_retain, data_decompose):
+def decompose_data(df, columns_retain, columns_decompose):
     df_decompose = pd.DataFrame(columns=['type', 'value'] + columns_retain)
 
     for index, row in df.iterrows():