update import place

open-compass · Sep 19, 2024 · ec2b73a · ec2b73a
1 parent 38a0f8b
commit ec2b73a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/opencompass/datasets/dingo.py b/opencompass/datasets/dingo.py
@@ -41,7 +41,8 @@ def load(path: str):
 
 @ICL_EVALUATORS.register_module()
 class DingoEvaluator(BaseEvaluator):
-    def __init__(self):
+
+    def score(self, origin_prompt: List, predictions: List) -> dict:
         try:
             from dingo.model.model import Model
             from dingo.io import InputArgs
@@ -50,7 +51,6 @@ def __init__(self):
             raise ModuleNotFoundError(
                 '=========== dingo register fail. please try: pip install dingo-python. ===========')
 
-    def score(self, origin_prompt: List, predictions: List) -> dict:
         current_time = time.strftime('%Y%m%d_%H%M%S', time.localtime())
         file_data = [{'prompt':pmt, 'prediction':prd} for pmt, prd in zip(origin_prompt, predictions)]
         file_name = 'dingo_file_' + current_time + '.jsonl'