Add pyupgrade pre-commit hook + modernize python code (#2611)

This commit is contained in:
Asher Foa
2025-06-10 14:52:38 -04:00
committed by GitHub
parent 272985f1bb
commit effd0c4911
18 changed files with 47 additions and 45 deletions

View File

@@ -21,13 +21,13 @@ class WorkflowRunResultRequest(BaseModel):
def load_webvoyager_case_from_json(file_path: str, group_id: str = "") -> Iterator[WebVoyagerTestCase]:
with open("evaluation/datasets/webvoyager_reference_answer.json", "r") as answer_file:
with open("evaluation/datasets/webvoyager_reference_answer.json") as answer_file:
webvoyager_answers: dict = json.load(answer_file)
if not group_id:
group_id = str(uuid4())
with open(file_path, "r", encoding="utf-8") as file:
with open(file_path, encoding="utf-8") as file:
for line in file:
test_case: dict[str, str] = json.loads(line)
web_name, id = test_case["id"].split("--")
@@ -47,7 +47,7 @@ def load_webvoyager_case_from_json(file_path: str, group_id: str = "") -> Iterat
def load_records_from_json(file_path: str) -> Iterator[WorkflowRunResultRequest]:
with open(file_path, "r", encoding="utf-8") as f:
with open(file_path, encoding="utf-8") as f:
for line in f:
item: dict[str, str] = json.loads(line)
id = item["id"]

View File

@@ -35,7 +35,7 @@ def main(
) -> None:
client = SkyvernClient(base_url=base_url, credentials=cred)
with open(record_json_path, "r", encoding="utf-8") as file:
with open(record_json_path, encoding="utf-8") as file:
with open(output_csv_path, newline="", mode="w", encoding="utf-8") as csv_file:
writer = csv.DictWriter(csv_file, fieldnames=csv_headers)
writer.writeheader()

View File

@@ -79,7 +79,7 @@ async def run_eval(
) -> None:
client = SkyvernClient(base_url=base_url, credentials=cred)
with open(record_json_path, "r", encoding="utf-8") as file:
with open(record_json_path, encoding="utf-8") as file:
with open(output_csv_path, newline="", mode="w", encoding="utf-8") as csv_file:
writer = csv.DictWriter(csv_file, fieldnames=csv_headers)
writer.writeheader()