Skip to content

Commit 268b903

Browse files
committed
fix: modify the function name
1 parent e82f8c0 commit 268b903

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

veadk/evaluation/base_evaluator.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -82,12 +82,12 @@ def __init__(
8282
self.result_list: list[EvalResultData] = []
8383
self.agent_information_list: list[dict] = []
8484

85-
def load_eval_set(self, eval_set_file: str) -> EvalSet:
85+
def _load_eval_set(self, eval_set_file: str) -> EvalSet:
8686
from .eval_set_file_loader import load_eval_set_from_file
8787

8888
return load_eval_set_from_file(eval_set_file)
8989

90-
def load_eval_set_from_tracing(self, tracing_file: str) -> EvalSet:
90+
def _load_eval_set_from_tracing(self, tracing_file: str) -> EvalSet:
9191
try:
9292
with open(tracing_file, "r") as f:
9393
tracing_data = json.load(f)
@@ -201,15 +201,15 @@ def generate_eval_data(self, file_path: str):
201201
raise ValueError(f"Error reading file {file_path}: {e}")
202202

203203
if isinstance(file_content, dict) and "eval_cases" in file_content:
204-
eval_cases = self.load_eval_set(file_path).eval_cases
204+
eval_cases = self._load_eval_set(file_path).eval_cases
205205
elif (
206206
isinstance(file_content, list)
207207
and len(file_content) > 0
208208
and all(
209209
isinstance(span, dict) and "trace_id" in span for span in file_content
210210
)
211211
):
212-
eval_cases = self.load_eval_set_from_tracing(file_path).eval_cases
212+
eval_cases = self._load_eval_set_from_tracing(file_path).eval_cases
213213
else:
214214
raise ValueError(
215215
f"Unsupported file format in {file_path}. Please provide a valid file."

0 commit comments

Comments
 (0)