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

---
 FlagEmbedding/evaluation/air_bench/__init__.py  | 4 +---
 FlagEmbedding/evaluation/air_bench/arguments.py | 1 +
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/FlagEmbedding/evaluation/air_bench/__init__.py b/FlagEmbedding/evaluation/air_bench/__init__.py
index e9ffb2ec..29c54943 100644
--- 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
index 465fe58e..1e924bca 100644
--- 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