diff --git a/dask_jobqueue/core.py b/dask_jobqueue/core.py index 4402e886..521829d4 100644 --- a/dask_jobqueue/core.py +++ b/dask_jobqueue/core.py @@ -142,7 +142,6 @@ class JobQueueCluster(Cluster): submit_command = None cancel_command = None scheduler_name = '' - _adaptive_options = {'worker_key': lambda ws: _job_id_from_worker_name(ws.name)} job_id_regexp = r'(?P\d+)' def __init__(self, @@ -258,6 +257,9 @@ def __repr__(self): total_workers, len(self.running_jobs), total_jobs) ) + def _worker_key(self, worker_state): + return _job_id_from_worker_name(worker_state.name) + @property def pending_jobs(self): """ Jobs pending in the queue """