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={}, ) ],