Skip to content

Commit 7056a72

Browse files
authored
Fix Modal unit tests following change in #382 that added a 'name' parameter (#387)
1 parent 47bca76 commit 7056a72

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

cubed/tests/runtime/test_modal_async.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -35,20 +35,22 @@
3535
@stub.function(
3636
image=image, secret=modal.Secret.from_name("my-aws-secret"), retries=2, timeout=10
3737
)
38-
def deterministic_failure_modal(i, path=None, timing_map=None):
39-
return deterministic_failure(path, timing_map, i)
38+
def deterministic_failure_modal(i, path=None, timing_map=None, *, name=None):
39+
return deterministic_failure(path, timing_map, i, name=name)
4040

4141

4242
@stub.function(image=image, secret=modal.Secret.from_name("my-aws-secret"), timeout=10)
43-
def deterministic_failure_modal_no_retries(i, path=None, timing_map=None):
44-
return deterministic_failure(path, timing_map, i)
43+
def deterministic_failure_modal_no_retries(i, path=None, timing_map=None, *, name=None):
44+
return deterministic_failure(path, timing_map, i, name=name)
4545

4646

4747
@stub.function(
4848
image=image, secret=modal.Secret.from_name("my-aws-secret"), retries=2, timeout=300
4949
)
50-
def deterministic_failure_modal_long_timeout(i, path=None, timing_map=None):
51-
return deterministic_failure(path, timing_map, i)
50+
def deterministic_failure_modal_long_timeout(
51+
i, path=None, timing_map=None, *, name=None
52+
):
53+
return deterministic_failure(path, timing_map, i, name=name)
5254

5355

5456
async def run_test(app_function, input, use_backups=False, batch_size=None, **kwargs):

0 commit comments

Comments
 (0)