fix bug in splitting output of react script

ottaviadipasquale · Jul 26, 2021 · 79c5f62 · 79c5f62
1 parent 212bfba
commit 79c5f62
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/react/__init__.py b/react/__init__.py
@@ -1,2 +1,2 @@
-__version__ = '0.1.5'
+__version__ = '0.1.6'
 from . import utils
diff --git a/script/react b/script/react
@@ -207,7 +207,11 @@ def main():
     logging.info(f'Saved result stage 2 in {fpath_result_stage1}')
 
     for i, f in enumerate(fpath_ic):
-        nib.save(nib.Nifti1Image(np.squeeze(beta2[..., i]),
+        if n_pet_maps == 1:
+            img = beta2
+        else:
+            img = beta2[..., i]
+        nib.save(nib.Nifti1Image(np.squeeze(img),
                                  affine=volume_pet.affine), f)
         logging.info(f'Saved result IC{i} of stage 2 in {f}')