change import place

opencompass/datasets/dingo.py

@@ -12,13 +12,6 @@
 
 from .base import BaseDataset
 
-try:
-    from dingo.model.model import Model
-    from dingo.io import InputArgs
-    from dingo.exec import Executor
-except Exception:
-    raise ModuleNotFoundError('=========== dingo register fail. please try: pip install dingo-python. ===========')
-
 @LOAD_DATASET.register_module()
 class DingoDataset(BaseDataset):
 
@@ -48,6 +41,14 @@ def load(path: str):
 
 @ICL_EVALUATORS.register_module()
 class DingoEvaluator(BaseEvaluator):
+    def __init__(self):
+        try:
+            from dingo.model.model import Model
+            from dingo.io import InputArgs
+            from dingo.exec import Executor
+        except Exception:
+            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())

requirements/extra.txt

@@ -17,4 +17,4 @@ pyext
 pypinyin
 # RULER
 wonderwords
-dingo
+dingo-python