Skip to content

Commit acd4886

Browse files
Update benchmark.py (#966)
1 parent d4baa67 commit acd4886

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

development/benchmark.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
def _get_image_and_predictor(model_type, device):
3434
example_data = fetch_livecell_example_data("../examples/data")
3535
image = imageio.imread(example_data)
36-
predictor = util.get_sam_model(device, model_type)
36+
predictor = util.get_sam_model(device=device, model_type=model_type)
3737
return image, predictor
3838

3939

0 commit comments

Comments
 (0)