Explorar o código

fix for new worker calc

Avik Datta %!s(int64=4) %!d(string=hai) anos
pai
achega
1423594a6c
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      dags/dag1_calculate_hpc_worker.py

+ 1 - 1
dags/dag1_calculate_hpc_worker.py

@@ -34,7 +34,7 @@ def get_new_workers(**kwargs):
     if 'ti' not in kwargs:
       raise ValueError('ti not present in kwargs')
     ti = kwargs.get('ti')
-    active_tasks = ti.xcom_pull(task_ids='qstat_on_remote')
+    active_tasks = ti.xcom_pull(task_ids='fetch_active_jobs_from_hpc')
     active_tasks = active_tasks.decode()
     active_tasks = json.loads(active_tasks)
     queued_tasks = ti.xcom_pull(task_ids='fetch_queue_list_from_redis')