diff --git a/agbenchmark/reports/ReportManager.py b/agbenchmark/reports/ReportManager.py index 18db144f..de8d95e7 100644 --- a/agbenchmark/reports/ReportManager.py +++ b/agbenchmark/reports/ReportManager.py @@ -63,7 +63,9 @@ class ReportManager: "command": command.split(os.sep)[-1], "benchmark_git_commit_sha": BENCHMARK_GIT_COMMIT_SHA, "agent_git_commit_sha": AGENT_GIT_COMMIT_SHA, - "completion_time": datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%S+00:00"), + "completion_time": datetime.now(timezone.utc).strftime( + "%Y-%m-%dT%H:%M:%S+00:00" + ), "benchmark_start_time": BENCHMARK_START_TIME, "metrics": { "run_time": str(round(time.time() - self.start_time, 2)) + " seconds", diff --git a/send_to_googledrive.py b/send_to_googledrive.py index aa9a6b09..e212e6bc 100644 --- a/send_to_googledrive.py +++ b/send_to_googledrive.py @@ -1,14 +1,13 @@ import base64 import json import os +import re import gspread import pandas as pd from dotenv import load_dotenv from oauth2client.service_account import ServiceAccountCredentials -import re - # Load environment variables from .env file load_dotenv() @@ -114,7 +113,7 @@ for agent_dir in os.listdir(base_dir): benchmark_start_time = data.get("benchmark_start_time", "") # Check if benchmark_start_time complies with the required format - pattern = re.compile(r'\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}\+00:00') + pattern = re.compile(r"\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}\+00:00") if not pattern.fullmatch(benchmark_start_time): continue # Skip processing this report if the date is not in the correct format # Loop through each test