Update niworkflows/viz/utils.py

Co-authored-by: Chris Markiewicz <[email protected]>

Author: mgxd

niworkflows/viz/utils.py

@@ -341,15 +341,16 @@ def plot_registration(
     if cuts is None:
         raise NotImplementedError  # TODO
 
-    anat_nii = _ensure_nii(anat_nii)
+    # nilearn 0.10.0 uses Nifti-specific methods
+    anat_nii = nb.Nifti1Image.from_image(anat_nii)
 
     out_files = []
     if estimate_brightness:
         plot_params = robust_set_limits(anat_nii.get_fdata().reshape(-1), plot_params)
 
     # FreeSurfer ribbon.mgz
     if contour:
-        contour = _ensure_nii(contour)
+        contour = nb.Nifti1Image.from_image(anat_nii)
 
     ribbon = contour is not None and np.array_equal(
         np.unique(contour.get_fdata()), [0, 2, 3, 41, 42]