diff --git a/platipy/imaging/label/fusion.py b/platipy/imaging/label/fusion.py index a6bc3d4a..de384750 100644 --- a/platipy/imaging/label/fusion.py +++ b/platipy/imaging/label/fusion.py @@ -12,10 +12,12 @@ # See the License for the specific language governing permissions and # limitations under the License. +from functools import reduce +import warnings + import numpy as np import SimpleITK as sitk -from functools import reduce from scipy.stats import pearsonr from skimage.util.shape import view_as_windows from platipy.imaging.registration.utils import smooth_and_resample diff --git a/platipy/imaging/projects/nnunet/run.py b/platipy/imaging/projects/nnunet/run.py index 9044fe56..4cb2e1dd 100644 --- a/platipy/imaging/projects/nnunet/run.py +++ b/platipy/imaging/projects/nnunet/run.py @@ -78,7 +78,7 @@ def download_and_install_nnunet_task(task_name, zip_url): try: from nnunet.inference.pretrained_models.download_pretrained_model import install_model_from_zip_file except ImportError: - raise ImportError("nnUNet Library not found. Be sure to install platipy with the required extras: 'pip install platipy[cardiac]' or 'pip install platipy[nnunet]'") + raise ImportError("nnUNet Library not found. Be sure to install platipy with the required extras: 'pip install platipy[cardiac]' or 'pip install platipy[nnunet]'") logger.info("Installing Task %s from %s", task_name, zip_url) temp_dir = tempfile.mkdtemp()