Fix Modal unit tests following change in #382 that added a 'name' par…

…ameter (#387)
cubed-dev · Feb 19, 2024 · 7056a72 · 7056a72
1 parent 47bca76
commit 7056a72
Showing 1 changed file with 8 additions and 6 deletions.
diff --git 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):