🐛 Fix output path for annotationstore output.

Author: shaneahmed

tests/engines/test_patch_predictor.py

@@ -490,7 +490,7 @@ def test_cli_model_single_file(sample_svs: Path, tmp_path: Path) -> None:
     )
 
     assert models_wsi_result.exit_code == 0
-    assert (tmp_path / "output" / "output.db").exists()
+    assert (tmp_path / "output" / (sample_svs.stem + ".db")).exists()
 
 
 def test_cli_model_multiple_file_mask(remote_sample: Callable, tmp_path: Path) -> None:

tiatoolbox/models/engine/engine_abc.py

@@ -673,9 +673,7 @@ def save_predictions(
         ) and output_type.lower() != "annotationstore":
             return processed_predictions
 
-        output_file = Path(kwargs.get("output_file", "output.db"))
-
-        save_path = save_dir / output_file
+        save_path = Path(kwargs.get("output_file", save_dir / "output.db"))
 
         if output_type.lower() == "annotationstore":
             # scale_factor set from kwargs
@@ -1063,6 +1061,7 @@ def _run_wsi_mode(
                 raw_predictions=raw_predictions,
                 **kwargs,
             )
+            kwargs["output_file"] = out[image]
             out[image] = self.save_predictions(
                 processed_predictions=processed_predictions,
                 output_type=output_type,