misc

scripts/convert_cls2fg.py

@@ -18,7 +18,6 @@ import argparse
 import shutil
 import sys
 from pathlib import Path
-from tokenize import PseudoExtras
 
 from hydra import initialize_config_module
 from loguru import logger

scripts/convert_seg2det_restore.py

@@ -21,9 +21,11 @@ from pathlib import Path
 from hydra import initialize_config_module
 
 from nndet.utils.config import compose
+from nndet.utils.check import env_guard
 
 
-if __name__ == '__main__':
+@env_guard
+def main():
     """
     Automatically deletes files generated by seg2det and restores
     the orignal segmentations
@@ -54,3 +56,7 @@ if __name__ == '__main__':
                     os.rename(f, f.parent / f"{f.name.rsplit('_', 1)[0]}.nii.gz")
             else:
                 print(f"{p} is not a dir. Skipping.")
+
+
+if __name__ == '__main__':
+    main()