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):