Gracefully die when attempting to run Brats dataset with torch>=2.2

Author: jan-grzybek-ampere

utils/cv/brats.py

@@ -50,6 +50,10 @@ def __init__(self, dataset_dir_path=None):
         self.__input_details = list()
 
         if not self.__preprocessed_dir_path.is_dir():
+            import pkg_resources
+            from packaging import version
+            if version.parse(pkg_resources.get_distribution("torch").version) >= version.parse("2.2"):
+                utils.print_goodbye_message_and_die("nnunet library requires torch<2.2")
             self.__preprocess()
 
         with open(os.path.join(Path(self.__preprocessed_dir_path, "preprocessed_files.pkl")), "rb") as f: