FIX: Ensure contour is also nii

mgxd · effigies · commit 9ed95624dc71 · 2023-01-13T16:52:07.000-05:00
diff --git a/niworkflows/viz/utils.py b/niworkflows/viz/utils.py
@@ -348,6 +348,9 @@ def plot_registration(
         plot_params = robust_set_limits(anat_nii.get_fdata().reshape(-1), plot_params)
 
     # FreeSurfer ribbon.mgz
+    if contour:
+        contour = _ensure_nii(contour)
+
     ribbon = contour is not None and np.array_equal(
         np.unique(contour.get_fdata()), [0, 2, 3, 41, 42]
     )

Original file line number	Diff line number	Diff line change
`@@ -348,6 +348,9 @@ def plot_registration(`
`348`	`348`	`plot_params = robust_set_limits(anat_nii.get_fdata().reshape(-1), plot_params)`
`349`	`349`
`350`	`350`	`# FreeSurfer ribbon.mgz`
	`351`	`+ if contour:`
	`352`	`+ contour = _ensure_nii(contour)`
	`353`	`+`
`351`	`354`	`ribbon = contour is not None and np.array_equal(`
`352`	`355`	`np.unique(contour.get_fdata()), [0, 2, 3, 41, 42]`
`353`	`356`	`)`