From 681b7d27cfc6f7d76e0d871169b4c73aca2592f0 Mon Sep 17 00:00:00 2001 From: ahuang11 Date: Wed, 4 Jan 2023 13:52:07 -0800 Subject: [PATCH 1/2] Add reqs --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 2fe1ef2..305542e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,3 @@ -prefect>=2.6.0 +prefect @ git+https://github.com/prefecthq/prefect.git@task_runner_test_suite_quote ray[default] >= 1.12.0; python_version >= '3.7' and python_version < '3.10' pickle5 >= 0.0.11; python_version < '3.8' From 9d026d66f1d6e428a5daa919ab121df893bd1897 Mon Sep 17 00:00:00 2001 From: ahuang11 Date: Wed, 4 Jan 2023 14:24:30 -0800 Subject: [PATCH 2/2] Fix throw exc --- tests/test_task_runners.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_task_runners.py b/tests/test_task_runners.py index 60ca96e..d5f0c4c 100644 --- a/tests/test_task_runners.py +++ b/tests/test_task_runners.py @@ -229,7 +229,7 @@ async def test_exception_to_crashed_state_in_flow_run( (raised_exception, state_exception_type) = exceptions async def throws_exception_before_task_begins( - task, task_run, parameters, wait_for, result_factory, settings + task, task_run, parameters, wait_for, result_factory, settings, **kwds ): """ Simulates an exception occurring while a remote task runner is attempting