from rdagent.core.conf import BenchmarkSettings
from rdagent.core.utils import import_class
from rdagent.benchmark.eval_method import FactorImplementEval
from rdagent.benchmark.data_process import load_eval_data

# 1.read the settings
bs = BenchmarkSettings()

# 2.read and prepare the eval_data
test_cases = load_eval_data(bs.bench_version)

# 3.declare the method to be tested and pass the arguments.
# TODO: Whether it is necessary to define two Eval method classes for two data type?
method_cls = import_class(bs.bench_method_cls)
generate_method = method_cls(bs.bench_method_extra_kwargs)

# 4.declare the eval method and pass the arguments.
eval_method = FactorImplementEval(
    method=generate_method,
    test_cases=test_cases,
    catch_eval_except=True,
    test_round=bs.bench_test_round,
)

# 5.run the eval
eval_method.eval()

# 6.save the result
eval_method.save(output_path = bs.bench_result_path)