backtracking from moving bgym.benchmarks etc

Author: TLSDC

src/agentlab/agents/agent_args.py

@@ -1,7 +1,5 @@
 import bgym
-from bgym import AbstractAgentArgs
-
-from agentlab.experiments.benchmark import Benchmark
+from bgym import AbstractAgentArgs, Benchmark
 
 
 class AgentArgs(AbstractAgentArgs):

src/agentlab/agents/dynamic_prompting.py

@@ -9,10 +9,10 @@
 from warnings import warn
 
 import bgym
+from bgym import HighLevelActionSetArgs
 from browsergym.core.action.base import AbstractActionSet
 from browsergym.utils.obs import flatten_axtree_to_str, flatten_dom_to_str, overlay_som, prune_html
 
-from agentlab.experiments.benchmark import HighLevelActionSetArgs
 from agentlab.llm.llm_utils import (
     BaseMessage,
     ParseError,

src/agentlab/agents/generic_agent/agent_configs.py

@@ -3,10 +3,10 @@
 """
 
 import bgym
+from bgym import HighLevelActionSetArgs
 
 from agentlab.agents import dynamic_prompting as dp
 from agentlab.experiments import args
-from agentlab.experiments.benchmark import HighLevelActionSetArgs
 from agentlab.llm.llm_configs import CHAT_MODEL_ARGS_DICT
 
 from .generic_agent import GenericAgentArgs

src/agentlab/agents/generic_agent/generic_agent.py

@@ -14,11 +14,11 @@
 from warnings import warn
 
 import bgym
+from bgym import Benchmark
 from browsergym.experiments.agent import Agent, AgentInfo
 
 from agentlab.agents import dynamic_prompting as dp
 from agentlab.agents.agent_args import AgentArgs
-from agentlab.experiments.benchmark import Benchmark
 from agentlab.llm.chat_api import BaseModelArgs
 from agentlab.llm.llm_utils import Discussion, ParseError, SystemMessage, retry
 from agentlab.llm.tracking import cost_tracker_decorator

src/agentlab/agents/generic_agent/reproducibility_agent.py

@@ -19,11 +19,11 @@
 from pathlib import Path
 
 import bgym
+from bgym import HighLevelActionSetArgs
 from browsergym.experiments.agent import AgentInfo
 from bs4 import BeautifulSoup
 
 from agentlab.agents.agent_args import AgentArgs
-from agentlab.experiments.benchmark import HighLevelActionSetArgs
 from agentlab.experiments.loop import ExpArgs, ExpResult, yield_all_exp_results
 from agentlab.experiments.study import Study
 from agentlab.llm.chat_api import make_assistant_message

src/agentlab/agents/visual_agent/agent_configs.py

@@ -1,7 +1,7 @@
 import bgym
+from bgym import HighLevelActionSetArgs
 
 import agentlab.agents.dynamic_prompting as dp
-from agentlab.experiments.benchmark import HighLevelActionSetArgs
 from agentlab.llm.llm_configs import CHAT_MODEL_ARGS_DICT
 
 from .visual_agent import VisualAgentArgs

src/agentlab/agents/visual_agent/visual_agent.py

@@ -11,11 +11,11 @@
 from dataclasses import asdict, dataclass
 
 import bgym
+from bgym import Benchmark
 from browsergym.experiments.agent import Agent, AgentInfo
 
 from agentlab.agents import dynamic_prompting as dp
 from agentlab.agents.agent_args import AgentArgs
-from agentlab.experiments.benchmark import Benchmark
 from agentlab.llm.chat_api import BaseModelArgs
 from agentlab.llm.llm_utils import Discussion, ParseError, SystemMessage, retry
 from agentlab.llm.tracking import cost_tracker_decorator