diff --git a/tests/integration/scoring/test_scoring.py b/tests/integration/scoring/test_scoring.py index 78a322d20..315ff050c 100644 --- a/tests/integration/scoring/test_scoring.py +++ b/tests/integration/scoring/test_scoring.py @@ -5,11 +5,9 @@ # the root directory of this source tree. -import os from pathlib import Path import pandas as pd - import pytest @@ -151,11 +149,7 @@ def test_scoring_score_with_aggregation_functions( df = pd.read_csv(Path(__file__).parent.parent / "datasets" / "test_dataset.csv") rows = df.to_dict(orient="records") - scoring_fns_list = [ - x - for x in llama_stack_client.scoring_functions.list() - if x.provider_id == provider_id - ] + scoring_fns_list = [x for x in llama_stack_client.scoring_functions.list() if x.provider_id == provider_id] if len(scoring_fns_list) == 0: pytest.skip(f"No scoring functions found for provider {provider_id}, skipping")