From 857acb24d6173aa8990c441071c04b9f5e789e9b Mon Sep 17 00:00:00 2001 From: Morgan Ekmefjord Date: Thu, 12 Dec 2024 18:09:14 +0100 Subject: [PATCH] Adding option to override where interim results are saved. --- ptp/inference/tasks.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/ptp/inference/tasks.py b/ptp/inference/tasks.py index ccfada4..fda2eb8 100644 --- a/ptp/inference/tasks.py +++ b/ptp/inference/tasks.py @@ -111,7 +111,10 @@ def run_inference(job_id): """ print("Concatenating results...", flush=True) - result_file_path = f"resultfilepath-{job_id}.csv" + + BASE_DIR = os.environ.get("BASE_DIR", "/app/") + os.makedirs(f"{BASE_DIR}/interim", exist_ok=True) + result_file_path = f"{BASE_DIR}/interim/resultfilepath-{job_id}.csv" try: # Paths for input and output