From cc64309e970d1f0d6fae84a9d76eaf0b6443e07d Mon Sep 17 00:00:00 2001 From: Adam Fourney Date: Thu, 6 Feb 2025 23:42:02 -0800 Subject: [PATCH] Fixed pyright. --- python/packages/agbench/src/agbench/run_cmd.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/python/packages/agbench/src/agbench/run_cmd.py b/python/packages/agbench/src/agbench/run_cmd.py index b7d250f7ae12..decc8b20a689 100644 --- a/python/packages/agbench/src/agbench/run_cmd.py +++ b/python/packages/agbench/src/agbench/run_cmd.py @@ -355,28 +355,28 @@ def replace_env_var(match: Any) -> str: pattern = re.compile(r"\$\{(\w+)\}") - def replace_in_dict(d: Dict[Any, Any]) -> None: + def replace_in_dict(d: Dict[str, Any]) -> None: for key, value in d.items(): if isinstance(value, str): d[key] = pattern.sub(replace_env_var, value) elif isinstance(value, dict): - replace_in_dict(value) + replace_in_dict(cast(Dict[str, Any], value)) elif isinstance(value, list): - replace_in_list(value) + replace_in_list(cast(List[Any], value)) def replace_in_list(lst: List[Any]) -> None: for i, item in enumerate(lst): if isinstance(item, str): lst[i] = pattern.sub(replace_env_var, item) elif isinstance(item, dict): - replace_in_dict(item) + replace_in_dict(cast(Dict[str, Any], item)) elif isinstance(item, list): - replace_in_list(item) + replace_in_list(cast(List[str], item)) if isinstance(json_data, dict): - replace_in_dict(json_data) + replace_in_dict(cast(Dict[str, Any], json_data)) elif isinstance(json_data, list): - replace_in_list(json_data) + replace_in_list(cast(List[Any], json_data)) def run_scenario_natively(work_dir: str, env: Mapping[str, str], timeout: int = TASK_TIMEOUT) -> None: