Reference internal norm template location

KatherLab · May 4, 2024 · 320bdaa · 320bdaa
1 parent fade593
commit 320bdaa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/stamp/cli.py b/stamp/cli.py
@@ -112,6 +112,7 @@ def run_cli(args: argparse.Namespace):
             )
             c = cfg.preprocessing
             # Some checks
+            normalization_template_path = Path(f"{os.environ['STAMP_RESOURCES_DIR']}/normalization_template.jpg")
             if c.norm and not Path(normalization_template_path).exists():
                 raise ConfigurationError(f"Normalization template {normalization_template_path} does not exist, please run `stamp setup` to download it.")
             if c.feat_extractor == 'ctp':