Merge pull request #1256 from hanhainebula/master

Fix air-bench eval bugs: AIRBenchEvalArgs
FlagOpen · Nov 22, 2024 · eb2025f · eb2025f
2 parents 03d5ab0 + 7bd8ad0
commit eb2025f
Showing 1 changed file with 3 additions and 1 deletion.
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__ = [