fix

Author: Dylan Huang

tests/pytest/test_pytest_ids.py

@@ -45,7 +45,7 @@ def eval_fn(row: EvaluationRow) -> EvaluationRow:
 
     # Manually invoke all parameter combinations within a single test
     for ds_path in dataset_paths:
-        await eval_fn(model="dummy/local-model", dataset_path=[ds_path])
+        await eval_fn(dataset_path=[ds_path], completion_params={"temperature": 0.0, "model": "dummy/local-model"})
 
     # Assertions on IDs generated by the decorator logic
     assert len(logger.read()) == 38
@@ -89,12 +89,15 @@ def eval_fn(row: EvaluationRow) -> EvaluationRow:
         "tests/pytest/data/markdown_dataset.jsonl",
         "tests/pytest/data/markdown_dataset.jsonl",
     ]
-    input_params_list = [{"temperature": 0.0}, {"temperature": 1.0}]
+    completion_params_list = [
+        {"temperature": 0.0, "model": "dummy/local-model"},
+        {"temperature": 1.0, "model": "dummy/local-model"},
+    ]
 
     # Manually invoke all parameter combinations within a single test
     for ds_path in dataset_paths:
-        for params in input_params_list:
-            await eval_fn(model="dummy/local-model", dataset_path=[ds_path], input_params=params)
+        for params in completion_params_list:
+            await eval_fn(dataset_path=[ds_path], completion_params=params)
 
     # Assertions on IDs generated by the decorator logic
     assert len(unique_invocation_ids) == 1