Skip to content

Commit 0e44e6c

Browse files
author
Dylan Huang
committed
fix
1 parent 847301f commit 0e44e6c

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

eval_protocol/pytest/utils.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313
DatasetPathParam,
1414
EvaluationInputParam,
1515
InputMessagesParam,
16-
InputRowsParam,
1716
RolloutProcessorConfig,
1817
)
1918
from eval_protocol.pytest.exception_config import ExceptionHandlerConfig, get_default_exception_handler_config
@@ -167,7 +166,7 @@ def generate_parameter_combinations(
167166
input_dataset: Optional[List[DatasetPathParam]],
168167
completion_params: List[CompletionParams],
169168
input_messages: Optional[List[InputMessagesParam]],
170-
input_rows: Optional[List[InputRowsParam]],
169+
input_rows: Optional[List[EvaluationRow]],
171170
evaluation_test_kwargs: Optional[List[EvaluationInputParam]],
172171
max_dataset_rows: Optional[int],
173172
combine_datasets: bool,

0 commit comments

Comments
 (0)