Adding option to override where interim results are saved.

pharmbio · Dec 12, 2024 · 857acb2 · 857acb2
1 parent ad1da3b
commit 857acb2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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