diff --git a/tests/providers/daskexecutor/test_dask_executor.py b/tests/providers/daskexecutor/test_dask_executor.py index 8e870cc4b84fe..9ba55bf421273 100644 --- a/tests/providers/daskexecutor/test_dask_executor.py +++ b/tests/providers/daskexecutor/test_dask_executor.py @@ -214,6 +214,7 @@ def test_dask_queues(self): assert success_future.done() assert success_future.exception() is None + @pytest.mark.execution_timeout(120) def test_dask_queues_no_queue_specified(self): self.cluster = LocalCluster(resources={"queue1": 1}) executor = DaskExecutor(cluster_address=self.cluster.scheduler_address) @@ -224,7 +225,7 @@ def test_dask_queues_no_queue_specified(self): success_future = next(k for k, v in executor.futures.items() if v == "success") # wait for the futures to execute, with a timeout - timeout = timezone.utcnow() + timedelta(seconds=30) + timeout = timezone.utcnow() + timedelta(seconds=100) while not success_future.done(): if timezone.utcnow() > timeout: raise ValueError(