diff --git a/tests/tests_integration/inference/test_model_inference.py b/tests/tests_integration/inference/test_model_inference.py index 4a7d7ec..850007f 100644 --- 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(