From 196a53bf0a4c4dd9395271cfd9a85aa5e2bdd94a Mon Sep 17 00:00:00 2001
From: hanhainebula <2512674094@qq.com>
Date: Fri, 22 Nov 2024 22:41:51 +0800
Subject: [PATCH] fix air-bench evaluation bug: AIRBenchEvalArgs

---
 FlagEmbedding/evaluation/air_bench/__init__.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/FlagEmbedding/evaluation/air_bench/__init__.py b/FlagEmbedding/evaluation/air_bench/__init__.py
index 29c54943..e9ffb2ec 100644
--- 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__ = [