add default out_path value

mpelchat04 · Sep 9, 2024 · 94f2dea · 94f2dea
1 parent 7e9d082
commit 94f2dea
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/inference_segmentation.py b/inference_segmentation.py
@@ -121,6 +121,7 @@ def main(params:Union[DictConfig, Dict]):
                             "classes_dict": classes_dict}
             with rasterio.open(mask_path, 'r+') as raster:
                 raster.update_tags(**meta_data_dict)
-        output_path = get_key_def('output_path', params['inference'], expected_type=str, to_path=True)
+        output_path = get_key_def('output_path', params['inference'], expected_type=str, to_path=True, 
+                                  default=str(working_folder / f"{aoi.aoi_id}_raw.tif"))
         os.rename(mask_path, output_path)
         logging.info(f"finished inferring image: {aoi.aoi_id} ")
diff --git a/tests/CI/test_gh_actions_ci.py b/tests/CI/test_gh_actions_ci.py
@@ -27,6 +27,7 @@ def test_ci(self) -> None:
                                              f"inference=default_{dataset_type}",
                                              f"training.num_gpus=0",
                                              f"loss={dataset_type}/lovasz",
+                                             "inference.output_path=./tst.tif",
                                              "hydra.job.num=0"],
                                   return_hydra_config=True)
                     hconf = HydraConfig()