Skip to content

Commit 5d5564d

Browse files
committed
fix air-bench evaluation bug: AIRBenchEvalArgs
1 parent 196a53b commit 5d5564d

2 files changed

Lines changed: 2 additions & 3 deletions

File tree

FlagEmbedding/evaluation/air_bench/__init__.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,4 @@
1-
from air_benchmark import EvalArgs as AIRBenchEvalArgs
2-
3-
from .arguments import AIRBenchEvalModelArgs
1+
from .arguments import AIRBenchEvalModelArgs, AIRBenchEvalArgs
42
from .runner import AIRBenchEvalRunner
53

64
__all__ = [

FlagEmbedding/evaluation/air_bench/arguments.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from dataclasses import dataclass, field
22
from typing import List, Optional
3+
from air_benchmark import EvalArgs as AIRBenchEvalArgs
34

45

56
@dataclass

0 commit comments

Comments
 (0)