From 79a13012fdd8a86f0184a01b6d424269ded9f798 Mon Sep 17 00:00:00 2001 From: Tom White Date: Mon, 29 Jul 2024 09:40:42 +0100 Subject: [PATCH] Fix Modal deprecation (show progress) --- cubed/runtime/executors/modal.py | 2 +- cubed/tests/runtime/test_modal.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/cubed/runtime/executors/modal.py b/cubed/runtime/executors/modal.py index 98be5510..68a5331b 100644 --- a/cubed/runtime/executors/modal.py +++ b/cubed/runtime/executors/modal.py @@ -209,7 +209,7 @@ async def async_execute_dag( ) -> None: if spec is not None: check_runtime_memory(spec) - async with app.run(): + async with app.run(show_progress=False): cloud = cloud or "aws" if cloud == "aws": app_function = run_remotely diff --git a/cubed/tests/runtime/test_modal.py b/cubed/tests/runtime/test_modal.py index 6e341709..2a9bf56d 100644 --- a/cubed/tests/runtime/test_modal.py +++ b/cubed/tests/runtime/test_modal.py @@ -65,7 +65,7 @@ def deterministic_failure_modal_long_timeout( async def run_test(app_function, input, use_backups=False, batch_size=None, **kwargs): outputs = set() - async with app.run(): + async with app.run(show_progress=False): async for output in map_unordered( app_function, input,