more black

Author: amanjaiswal73892

experiments/run_osworld.py

@@ -2,11 +2,9 @@
 import logging
 import os
 
-from tapeagents import agent
-
-from agentlab.agents.tool_use_agent.tool_use_agent import OSWORLD_CLAUDE, OSWORLD_OAI
+from agentlab.agents.tool_use_agent.tool_use_agent import OSWORLD_CLAUDE
 from agentlab.benchmarks.osworld import OsworldBenchmark
-from agentlab.experiments.study import make_study, Study
+from agentlab.experiments.study import Study, make_study
 
 fmt = "%(asctime)s - %(levelname)s - %(name)s:%(lineno)d - %(funcName)s() - %(message)s"
 logging.basicConfig(level=logging.INFO, force=True, format=fmt, handlers=[logging.StreamHandler()])
@@ -20,6 +18,7 @@ def get_most_recent_incomplete_study() -> Study:
     study.find_incomplete(include_errors=True)
     return study
 
+
 def get_task_ids() -> set[str]:
     with open("experiments/osworld_debug_task_ids.json", "r") as f:
         task_ids = json.load(f)
@@ -32,13 +31,15 @@ def main():
     relaunch = True
     agent_args = [
         OSWORLD_CLAUDE,
-                #    OSWORLD_OAI # performs poorly. 
-                   ]  # type: ignore
+        #    OSWORLD_OAI # performs poorly.
+    ]  # type: ignore
     parallel_backend = "ray"
     os.environ["AGENTLAB_DEBUG"] = os.environ.get("AGENTLAB_DEBUG", "1")
 
     study = make_study(
-        benchmark=OsworldBenchmark(test_set_name="test_small.json"), # or test_all.json (Exper)  # type: ignore
+        benchmark=OsworldBenchmark(
+            test_set_name="test_small.json"
+        ),  # or test_all.json (Exper)  # type: ignore
         agent_args=agent_args,  # type: ignore
         comment="osworld debug 2",
         logging_level=logging.INFO,

src/agentlab/benchmarks/osworld.py

@@ -733,7 +733,7 @@ def model_post_init(self, __context: Any) -> None:
 
     def fix_settings_file_path_in_config(self, task: dict) -> dict:
         """Fix the settings file path in the task configuration.
-        
+
         Args:
             task: Task configuration dictionary.
 

tests/benchmarks/test_osworld.py

@@ -1,4 +1,3 @@
-
 import tempfile
 from pathlib import Path
 from unittest.mock import patch
@@ -12,7 +11,6 @@
 )
 
 
-
 def mock_task_config() -> dict:
     """Mock task configuration for testing."""
     return {
@@ -282,4 +280,3 @@ def test_convert_agentlab_action_to_computer_13(self):
                 result = gym.convert_agentlab_action_to_computer_13('typing(text="hello")')
                 expected = {"action_type": "TYPING", "parameters": {"text": "hello"}}
                 assert result == expected
-