minor fix

OnnoKampman · May 8, 2024 · 8c72c46 · 8c72c46
1 parent 36c60ca
commit 8c72c46
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/benchmarks/fmri/sim/imputation_benchmark/run_statistical_test_between_estimation_methods.py b/benchmarks/fmri/sim/imputation_benchmark/run_statistical_test_between_estimation_methods.py
@@ -34,12 +34,12 @@
             all_method_significances_df = pd.DataFrame(
                 np.nan,
                 index=models_list,
-                columns=models_list
+                columns=models_list,
             )
             all_method_cod_df = pd.DataFrame(
                 np.nan,
                 index=models_list,
-                columns=models_list
+                columns=models_list,
             )
             for (i_model, j_model) in get_all_lower_triangular_indices_tuples(len(models_list)):
                 print(i_model, j_model)
@@ -54,10 +54,11 @@
                         logging.warning(
                             f"Test likelihoods not found: '{test_likelihoods_savepath:s}'"
                         )
+                        all_methods_test_likelihoods.append(np.full([num_trials], np.nan))
                         continue
                     test_likelihoods_df = pd.read_csv(
                         test_likelihoods_savepath,
-                        index_col=0
+                        index_col=0,
                     )
                     print(test_likelihoods_df)
                     logging.info(
@@ -89,6 +90,6 @@
             all_method_cod_df.to_csv(
                 os.path.join(test_likelihoods_savedir, likelihoods_cod_filename),
                 index=True,
-                float_format='%.3f'
+                float_format='%.3f',
             )
             logging.info(f"Saved {data_split:s} likelihoods Cohen's D '{likelihoods_cod_filename:s}' in '{test_likelihoods_savedir:s}'.")