From 7056a72da6e8ab0f7005230a255a843786e05c16 Mon Sep 17 00:00:00 2001 From: Tom White Date: Mon, 19 Feb 2024 13:53:42 +0000 Subject: [PATCH] Fix Modal unit tests following change in #382 that added a 'name' parameter (#387) --- cubed/tests/runtime/test_modal_async.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/cubed/tests/runtime/test_modal_async.py b/cubed/tests/runtime/test_modal_async.py index 88767b19..d25ff532 100644 --- a/cubed/tests/runtime/test_modal_async.py +++ b/cubed/tests/runtime/test_modal_async.py @@ -35,20 +35,22 @@ @stub.function( image=image, secret=modal.Secret.from_name("my-aws-secret"), retries=2, timeout=10 ) -def deterministic_failure_modal(i, path=None, timing_map=None): - return deterministic_failure(path, timing_map, i) +def deterministic_failure_modal(i, path=None, timing_map=None, *, name=None): + return deterministic_failure(path, timing_map, i, name=name) @stub.function(image=image, secret=modal.Secret.from_name("my-aws-secret"), timeout=10) -def deterministic_failure_modal_no_retries(i, path=None, timing_map=None): - return deterministic_failure(path, timing_map, i) +def deterministic_failure_modal_no_retries(i, path=None, timing_map=None, *, name=None): + return deterministic_failure(path, timing_map, i, name=name) @stub.function( image=image, secret=modal.Secret.from_name("my-aws-secret"), retries=2, timeout=300 ) -def deterministic_failure_modal_long_timeout(i, path=None, timing_map=None): - return deterministic_failure(path, timing_map, i) +def deterministic_failure_modal_long_timeout( + i, path=None, timing_map=None, *, name=None +): + return deterministic_failure(path, timing_map, i, name=name) async def run_test(app_function, input, use_backups=False, batch_size=None, **kwargs):