diff --git a/tests/unit/test_dataset_schema.py b/tests/unit/test_dataset_schema.py index f908609c1..d8d3b9af7 100644 --- a/tests/unit/test_dataset_schema.py +++ b/tests/unit/test_dataset_schema.py @@ -129,7 +129,13 @@ def test_single_type_evaluation_dataset(eval_sample): with pytest.raises(ValueError) as exc_info: EvaluationDataset(samples=[single_turn_sample, multi_turn_sample]) - assert "All samples must be of the same type" in str(exc_info.value) + error_message = str(exc_info.value) + + assert ( + "Sample at index 1 is of type " + in error_message + ) + assert "expected " in error_message def test_base_eval_sample():