fix inference integration test

Open-EO · Sep 27, 2024 · 7b654f5 · 7b654f5
1 parent 69cc449
commit 7b654f5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/tests_integration/inference/test_model_inference.py b/tests/tests_integration/inference/test_model_inference.py
@@ -60,7 +60,7 @@ def test_onnx_inference_local():
     assert output.shape == (1, 256, 256)
     assert len(np.unique(output.values)) == 3
 
-    output_path = Path(__file__).parent / "results/test_onnx_inference_local.nc"
+    output_path = Path(__file__).parent.parent / "results/test_onnx_inference_local.nc"
     print(output_path)
     output.to_netcdf(output_path)
 
@@ -113,7 +113,7 @@ def test_onnx_inference():
         ],
     )
 
-    output_path = Path(__file__).parent / "results/test_onnx_inference.tif"
+    output_path = Path(__file__).parent.parent / "results/test_onnx_inference.tif"
 
     # Download the results as tif file.
     job = cube.create_job(