From 8a0c1f330562a7d226a32a3d11517a653effa754 Mon Sep 17 00:00:00 2001
From: Barbara Bertozzi <barbara.bertozzi@psi.ch>
Date: Fri, 22 Aug 2025 18:15:39 +0200
Subject: [PATCH] feat: add TMPDIR for temporary files in cluster jobs

---
 src/sp2xr/helpers.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/sp2xr/helpers.py b/src/sp2xr/helpers.py
index 5f823a9..6ee379f 100644
--- a/src/sp2xr/helpers.py
+++ b/src/sp2xr/helpers.py
@@ -97,14 +97,17 @@ def initialize_cluster(config):
 
 
 def make_slurm_cluster(config):
+    tmpdir = os.environ.get("TMPDIR", "/tmp")
     cluster = SLURMCluster(
         cores=config["cluster"]["cores"],
         processes=config["cluster"]["cores"],
         memory=config["cluster"]["memory"],
         walltime=config["cluster"]["walltime"],
+        local_directory=tmpdir,
         job_extra_directives=[
             f"--partition={config['cluster']['partition']}",
             f"--output={config['cluster']['log_dir']}/slurm-%j.out",
+            f"--export=ALL,TMPDIR={tmpdir}",
         ],
     )
     client = Client(cluster)