From ecf1932424343522586fc5ee3cae296260cfa8ab Mon Sep 17 00:00:00 2001
From: Xi Yan <xiyan@meta.com>
Date: Mon, 11 Nov 2024 22:41:15 -0500
Subject: [PATCH] config rename

---
 llama_stack/providers/inline/scoring/basic/__init__.py | 4 ++--
 llama_stack/providers/inline/scoring/basic/config.py   | 4 ++--
 llama_stack/providers/inline/scoring/basic/scoring.py  | 4 ++--
 llama_stack/providers/registry/scoring.py              | 2 +-
 4 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/llama_stack/providers/inline/scoring/basic/__init__.py b/llama_stack/providers/inline/scoring/basic/__init__.py
index 002f74e86..ef04ece98 100644
--- a/llama_stack/providers/inline/scoring/basic/__init__.py
+++ b/llama_stack/providers/inline/scoring/basic/__init__.py
@@ -7,11 +7,11 @@ from typing import Dict
 
 from llama_stack.distribution.datatypes import Api, ProviderSpec
 
-from .config import MetaReferenceScoringConfig
+from .config import BasicScoringConfig
 
 
 async def get_provider_impl(
-    config: MetaReferenceScoringConfig,
+    config: BasicScoringConfig,
     deps: Dict[Api, ProviderSpec],
 ):
     from .scoring import MetaReferenceScoringImpl
diff --git a/llama_stack/providers/inline/scoring/basic/config.py b/llama_stack/providers/inline/scoring/basic/config.py
index bd4dcb9f0..d9dbe71bc 100644
--- a/llama_stack/providers/inline/scoring/basic/config.py
+++ b/llama_stack/providers/inline/scoring/basic/config.py
@@ -3,7 +3,7 @@
 #
 # This source code is licensed under the terms described in the LICENSE file in
 # the root directory of this source tree.
-from llama_stack.apis.scoring import *  # noqa: F401, F403
+from pydantic import BaseModel
 
 
-class MetaReferenceScoringConfig(BaseModel): ...
+class BasicScoringConfig(BaseModel): ...
diff --git a/llama_stack/providers/inline/scoring/basic/scoring.py b/llama_stack/providers/inline/scoring/basic/scoring.py
index ff1ee6c43..d346ebe35 100644
--- a/llama_stack/providers/inline/scoring/basic/scoring.py
+++ b/llama_stack/providers/inline/scoring/basic/scoring.py
@@ -14,7 +14,7 @@ from llama_stack.apis.datasets import *  # noqa: F403
 from llama_stack.apis.inference.inference import Inference
 from llama_stack.providers.datatypes import ScoringFunctionsProtocolPrivate
 
-from .config import MetaReferenceScoringConfig
+from .config import BasicScoringConfig
 from .scoring_fn.equality_scoring_fn import EqualityScoringFn
 from .scoring_fn.llm_as_judge_scoring_fn import LlmAsJudgeScoringFn
 from .scoring_fn.regex_parser_scoring_fn import RegexParserScoringFn
@@ -28,7 +28,7 @@ LLM_JUDGE_FNS = [LlmAsJudgeScoringFn]
 class MetaReferenceScoringImpl(Scoring, ScoringFunctionsProtocolPrivate):
     def __init__(
         self,
-        config: MetaReferenceScoringConfig,
+        config: BasicScoringConfig,
         datasetio_api: DatasetIO,
         datasets_api: Datasets,
         inference_api: Inference,
diff --git a/llama_stack/providers/registry/scoring.py b/llama_stack/providers/registry/scoring.py
index 1787cac77..ccf303d59 100644
--- a/llama_stack/providers/registry/scoring.py
+++ b/llama_stack/providers/registry/scoring.py
@@ -16,7 +16,7 @@ def available_providers() -> List[ProviderSpec]:
             provider_type="meta-reference",
             pip_packages=[],
             module="llama_stack.providers.inline.scoring.basic",
-            config_class="llama_stack.providers.inline.scoring.basic.MetaReferenceScoringConfig",
+            config_class="llama_stack.providers.inline.scoring.basic.BasicScoringConfig",
             api_dependencies=[
                 Api.datasetio,
                 Api.datasets,