Change remove to pop in DebugExecutor (#8073)

Due to difference in types between v1.10 and master we have to change set.remove to dict.pop method (cherry picked from commit 7393e3b693f06c81ec2d6fb818600b48146252f1)
apache · Apr 2, 2020 · 5e7d5f1 · 5e7d5f1
1 parent 895edc1
commit 5e7d5f1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/airflow/executors/debug_executor.py b/airflow/executors/debug_executor.py
@@ -145,5 +145,5 @@ def terminate(self):
 
     def change_state(self, key, state):
         self.log.debug("Popping %s from executor task queue.", key)
-        self.running.remove(key)
+        self.running.pop(key)
         self.event_buffer[key] = state
diff --git a/tests/executors/test_debug_executor.py b/tests/executors/test_debug_executor.py
@@ -44,7 +44,7 @@ def test_run_task(self, task_instance_mock):
         task_instance_mock.job_id = job_id
 
         executor = DebugExecutor()
-        executor.running = set([ti_key])
+        executor.running = {ti_key: task_instance_mock}
         succeeded = executor._run_task(task_instance_mock)
 
         assert succeeded
@@ -85,7 +85,7 @@ def test_end(self):
 
         executor = DebugExecutor()
         executor.tasks_to_run = [ti]
-        executor.running = set([ti.key])
+        executor.running = {ti.key: mock.MagicMock}
         executor.end()
 
         ti.set_state.assert_called_once_with(State.UPSTREAM_FAILED)