From bb54917de6c1e47037dbd29d42bd2ab01d798d08 Mon Sep 17 00:00:00 2001 From: Amogh Desai Date: Wed, 12 Feb 2025 17:19:44 +0530 Subject: [PATCH] AIP-72: Use runtime TI for listener tests --- task_sdk/tests/execution_time/test_task_runner.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/task_sdk/tests/execution_time/test_task_runner.py b/task_sdk/tests/execution_time/test_task_runner.py index b882aca2e9191..fcf2f3c17f6ac 100644 --- a/task_sdk/tests/execution_time/test_task_runner.py +++ b/task_sdk/tests/execution_time/test_task_runner.py @@ -1504,7 +1504,7 @@ def execute(self, context): log = mock.MagicMock() state, _ = run(runtime_ti, log) - finalize(ti, state, log) + finalize(runtime_ti, state, log) assert listener.state == [TaskInstanceState.RUNNING, TaskInstanceState.SUCCESS] @@ -1542,6 +1542,6 @@ def execute(self, context): log = mock.MagicMock() state, _ = run(runtime_ti, log) - finalize(ti, state, log) + finalize(runtime_ti, state, log) assert listener.state == [TaskInstanceState.RUNNING, TaskInstanceState.FAILED]