diff --git a/jale/core/analyses/contrast.py b/jale/core/analyses/contrast.py index d5cd232..abb6e6f 100644 --- a/jale/core/analyses/contrast.py +++ b/jale/core/analyses/contrast.py @@ -75,9 +75,14 @@ def contrast( ) else: logger.info(f"{meta_names[0]} x {meta_names[1]} - Computing positive contrast.") - main_effect1 = nb.loadsave.load( - project_path / f"MainEffect/Volumes/{meta_names[0]}_{correction_method}.nii" - ).get_fdata() + try: + main_effect1 = nb.loadsave.load( + project_path / f"MainEffect/Volumes/{meta_names[0]}_{correction_method}.nii" + ).get_fdata() + except FileNotFoundError: + main_effect1 = nb.loadsave.load( + project_path / f"MainEffect/Volumes/{meta_names[0]}_{correction_method}_empty.nii" + ).get_fdata() significance_mask1 = main_effect1 > 0 if significance_mask1.sum() > 0: stacked_masked_ma = np.vstack( @@ -102,9 +107,14 @@ def contrast( z1, sig_idxs1 = [], [] logger.info(f"{meta_names[1]} x {meta_names[0]} - Computing negative contrast.") - main_effect2 = nb.loadsave.load( - project_path / f"MainEffect/Volumes/{meta_names[1]}_{correction_method}.nii" - ).get_fdata() # type: ignore + try: + main_effect2 = nb.loadsave.load( + project_path / f"MainEffect/Volumes/{meta_names[1]}_{correction_method}.nii" + ).get_fdata() # type: ignore + except FileNotFoundError: + main_effect2 = nb.loadsave.load( + project_path / f"MainEffect/Volumes/{meta_names[1]}_{correction_method}_empty.nii" + ).get_fdata() significance_mask2 = main_effect2 > 0 if significance_mask2.sum() > 0: stacked_masked_ma = np.vstack(