Traceback (most recent call last):


 "/home/pieter/.venv/bin/nnUNet_predict", line 11, in <module>

load_entry_point('nnunet', 'console_scripts', 'nnUNet_predict')()


 "/home/pieter/unetcart/nnUNet/nnunet/inference/predict_simple.py", line 217, in main

predict_from_folder(model_folder_name, input_folder, output_folder, folds, save_npz, num_threads_preprocessing,


 "/home/pieter/unetcart/nnUNet/nnunet/inference/predict.py", line 630, in predict_from_folder

return predict_cases(model, list_of_lists[part_id::num_parts], output_files[part_id::num_parts], folds,


 "/home/pieter/unetcart/nnUNet/nnunet/inference/predict.py", line 182, in predict_cases

trainer, params = load_model_and_checkpoint_files(model, folds, mixed_precision=mixed_precision, checkpoint_name=checkpoint_name)


 "/home/pieter/unetcart/nnUNet/nnunet/training/model_restore.py", line 140, in load_model_and_checkpoint_files

trainer = restore_model(join(folds[0], "%s.model.pkl" % checkpoint_name), fp16=mixed_precision)


 "/home/pieter/unetcart/nnUNet/nnunet/training/model_restore.py", line 60, in restore_model

tr = recursive_find_python_class([search_in], name, current_module="nnunet.training.network_training")


 "/home/pieter/unetcart/nnUNet/nnunet/training/model_restore.py", line 37, in recursive_find_python_class

tr = recursive_find_python_class([join(folder[0], modname)], trainer_name, current_module=next_current_module)


 "/home/pieter/unetcart/nnUNet/nnunet/training/model_restore.py", line 37, in recursive_find_python_class

tr = recursive_find_python_class([join(folder[0], modname)], trainer_name, current_module=next_current_module)


 "/home/pieter/unetcart/nnUNet/nnunet/training/model_restore.py", line 28, in recursive_find_python_class

m = importlib.import_module(current_module + "." + modname)


 "/usr/lib/python3.8/importlib/__init__.py", line 127, in import_module

return _bootstrap._gcd_import(name[level:], package, level)


 "<frozen importlib._bootstrap>", line 1014, in _gcd_import


 "<frozen importlib._bootstrap>", line 991, in _find_and_load


 "<frozen importlib._bootstrap>", line 975, in _find_and_load_unlocked


 "<frozen importlib._bootstrap>", line 671, in _load_unlocked


 "<frozen importlib._bootstrap_external>", line 783, in exec_module


 "<frozen importlib._bootstrap>", line 219, in _call_with_frames_removed


 "/home/pieter/unetcart/nnUNet/nnunet/training/network_training/competitions_with_custom_Trainers/BraTS2020/nnUNetTrainerV2BraTSRegions_moreDA.py", line 19, in <module>

from nnunet.training.data_augmentation.data_augmentation_insaneDA2 import get_insaneDA_augmentation2


 "/home/pieter/unetcart/nnUNet/nnunet/training/data_augmentation/data_augmentation_insaneDA2.py", line 16, in <module>

from batchgenerators.transforms import DataChannelSelectionTransform, SegChannelSelectionTransform, SpatialTransform, \
ImportError: cannot import name 'SpatialTransform_2' from 'batchgenerators.transforms' (/home/pieter/.venv/lib/python3.8/site-packages/batchgenerators/transforms/__init__.py)
Traceback (most recent call last):


 "train.py", line 8, in <module>

from test import evaluate
ImportError: cannot import name 'evaluate'
