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)