Merge pull request #1257 from hanhainebula/master

Fix air-bench eval bugs: AIRBenchEvalArgs
FlagOpen · Nov 22, 2024 · a719aaa · a719aaa
2 parents eb2025f + 434f325
commit a719aaa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/FlagEmbedding/evaluation/air_bench/__init__.py b/FlagEmbedding/evaluation/air_bench/__init__.py
@@ -1,6 +1,4 @@
-from air_benchmark import EvalArgs as AIRBenchEvalArgs
-
-from .arguments import AIRBenchEvalModelArgs
+from .arguments import AIRBenchEvalModelArgs, AIRBenchEvalArgs
 from .runner import AIRBenchEvalRunner
 
 __all__ = [

diff --git a/FlagEmbedding/evaluation/air_bench/arguments.py b/FlagEmbedding/evaluation/air_bench/arguments.py
@@ -1,5 +1,6 @@
 from dataclasses import dataclass, field
 from typing import List, Optional
+from air_benchmark import EvalArgs as AIRBenchEvalArgs
 
 
 @dataclass