diff --git a/browser_env/actions.py b/browser_env/actions.py index 04ed355..d053b06 100644 --- a/browser_env/actions.py +++ b/browser_env/actions.py @@ -1378,7 +1378,9 @@ def parse_playwright_code(code: str) -> list[ParsedPlaywrightCode]: if isinstance(node, ast.Call): function_name = node.func.id # type: ignore[attr-defined] arguments = [ - ast.literal_eval(arg) if isinstance(arg, ast.Str) else arg + str(ast.literal_eval(arg)) + if isinstance(arg, ast.Str) + else str(arg) for arg in node.args ] keywords = { diff --git a/evaluation_harness/evaluators.py b/evaluation_harness/evaluators.py index d0417d4..8cad81a 100644 --- a/evaluation_harness/evaluators.py +++ b/evaluation_harness/evaluators.py @@ -10,7 +10,7 @@ from pathlib import Path from typing import Any, Tuple, Union from beartype import beartype -from nltk.tokenize import word_tokenize # type: ignore +from nltk.tokenize import word_tokenize from playwright.sync_api import CDPSession, Page from browser_env.actions import Action diff --git a/llms/providers/hf_utils.py b/llms/providers/hf_utils.py index b5e8987..83a5325 100644 --- a/llms/providers/hf_utils.py +++ b/llms/providers/hf_utils.py @@ -1,4 +1,4 @@ -from text_generation import Client # type: ignore +from text_generation import Client def generate_from_huggingface_completion( diff --git a/llms/tokenizers.py b/llms/tokenizers.py index 8e45ccf..1f6f338 100644 --- a/llms/tokenizers.py +++ b/llms/tokenizers.py @@ -1,7 +1,7 @@ from typing import Any import tiktoken -from transformers import LlamaTokenizer # type: ignore +from transformers import LlamaTokenizer class Tokenizer(object): diff --git a/tests/test_browser_env/test_script_browser_env.py b/tests/test_browser_env/test_script_browser_env.py index bf24734..2c6b401 100644 --- a/tests/test_browser_env/test_script_browser_env.py +++ b/tests/test_browser_env/test_script_browser_env.py @@ -123,7 +123,7 @@ def test_parallel_script_browser_env() -> None: # assert is_bearable(info["page"].tolist(), list[DetachedPage]) assert info["page"][0].url == "https://www.rfc-editor.org/rfc/rfc2606.html" assert info["page"][1].url == "https://www.rfc-editor.org/rfc/rfc6761.html" - vector_env.close() # type: ignore[no-untyped-call] + vector_env.close() def test_focus_placeholder_and_label(