fix air-bench evaluation bug: AIRBenchEvalArgs

hanhainebula · hanhainebula · commit 196a53bf0a4c · 2024-11-22T22:41:51.000+08:00
diff --git a/FlagEmbedding/evaluation/air_bench/__init__.py b/FlagEmbedding/evaluation/air_bench/__init__.py
@@ -1,4 +1,6 @@
-from .arguments import AIRBenchEvalModelArgs, AIRBenchEvalArgs
+from air_benchmark import EvalArgs as AIRBenchEvalArgs
+
+from .arguments import AIRBenchEvalModelArgs
 from .runner import AIRBenchEvalRunner
 
 __all__ = [
