diff --git a/evaluation/benchmarks/gaia/run_infer.py b/evaluation/benchmarks/gaia/run_infer.py index cc21cfc25d..f9841b9566 100644 --- a/evaluation/benchmarks/gaia/run_infer.py +++ b/evaluation/benchmarks/gaia/run_infer.py @@ -10,7 +10,6 @@ import huggingface_hub import pandas as pd from datasets import load_dataset from PIL import Image -from pydantic import SecretStr from evaluation.benchmarks.gaia.scorer import question_scorer from evaluation.benchmarks.gaia.utils import ( @@ -80,8 +79,7 @@ def get_config( config_copy = copy.deepcopy(config) load_from_toml(config_copy) - if config_copy.search_api_key: - config.search_api_key = SecretStr(config_copy.search_api_key) + config.search_api_key = config_copy.search_api_key return config