From a15a4a4be2fb823a579e33ab11a424f079d8b04c Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Tue, 12 Nov 2024 10:03:03 -0500 Subject: [PATCH] rebase --- .../providers/inline/scoring/braintrust/braintrust.py | 1 - llama_stack/providers/tests/scoring/fixtures.py | 6 +++--- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/llama_stack/providers/inline/scoring/braintrust/braintrust.py b/llama_stack/providers/inline/scoring/braintrust/braintrust.py index 0a21bc99c..973232f4e 100644 --- a/llama_stack/providers/inline/scoring/braintrust/braintrust.py +++ b/llama_stack/providers/inline/scoring/braintrust/braintrust.py @@ -64,7 +64,6 @@ class BraintrustScoringImpl(Scoring, ScoringFunctionsProtocolPrivate): async def validate_scoring_input_dataset_schema(self, dataset_id: str) -> None: dataset_def = await self.datasets_api.get_dataset(dataset_id=dataset_id) - print(dataset_def) if not dataset_def.schema or len(dataset_def.schema) == 0: raise ValueError( f"Dataset {dataset_id} does not have a schema defined. Please define a schema for the dataset." diff --git a/llama_stack/providers/tests/scoring/fixtures.py b/llama_stack/providers/tests/scoring/fixtures.py index 5b28dcfa8..14095b526 100644 --- a/llama_stack/providers/tests/scoring/fixtures.py +++ b/llama_stack/providers/tests/scoring/fixtures.py @@ -24,7 +24,7 @@ def scoring_basic() -> ProviderFixture: providers=[ Provider( provider_id="basic", - provider_type="basic", + provider_type="inline::basic", config={}, ) ], @@ -37,7 +37,7 @@ def scoring_braintrust() -> ProviderFixture: providers=[ Provider( provider_id="braintrust", - provider_type="braintrust", + provider_type="inline::braintrust", config={}, ) ], @@ -50,7 +50,7 @@ def scoring_llm_as_judge() -> ProviderFixture: providers=[ Provider( provider_id="llm-as-judge", - provider_type="llm-as-judge", + provider_type="inline::llm-as-judge", config={}, ) ],