Update dask controller for new dask_jobqueue

flatironinstitute · Feb 6, 2023 · 517fcee · 517fcee
1 parent 8172f8e
commit 517fcee
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/inferelator/distributed/dask_cluster_controller.py b/inferelator/distributed/dask_cluster_controller.py
@@ -192,10 +192,10 @@ def connect(cls, *args, **kwargs):
                 cores=cls._job_n_workers * cls._worker_n_threads,
                 processes=cls._job_n_workers,
                 job_mem=cls._job_mem,
-                env_extra=cls._config_env(),
+                job_script_prologue=cls._config_env(),
                 local_directory=cls._local_directory,
                 memory=cls._job_mem,
-                job_extra=cls._job_slurm_commands,
+                job_extra_directives=cls._job_slurm_commands,
                 job_cls=SLURMJobNoMemLimit,
                 **kwargs
             )

diff --git a/requirements-multiprocessing.txt b/requirements-multiprocessing.txt
@@ -1,3 +1,3 @@
 dask[complete]
-dask_jobqueue
+dask_jobqueue >= 0.8.0
 joblib