fix tests

Author: recursix

src/agentlab/agents/dynamic_prompting.py

@@ -385,7 +385,7 @@ def _prompt(self) -> str:
     URL: {page_url}
 """
             prompt_pieces.append(prompt_piece)
-        return  "\n".join(prompt_pieces)
+        return "\n".join(prompt_pieces)
 
 
 class Observation(Shrinkable):

src/agentlab/experiments/study.py

@@ -254,7 +254,7 @@ def get_report(self, ignore_cache=False, ignore_stale=False):
         return inspect_results.get_study_summary(
             self.dir, ignore_cache=ignore_cache, ignore_stale=ignore_stale
         )
-    
+
     def override_max_steps(self, max_steps):
         for exp_args in self.exp_args_list:
             exp_args.env_args.max_steps = max_steps

tests/agents/test_generic_prompt.py

@@ -32,15 +32,18 @@
 }
 
 OBS_HISTORY = [
-    base_obs | {
+    base_obs
+    | {
         "pruned_html": html_template.format(1),
         "last_action_error": "",
     },
-    base_obs | {
+    base_obs
+    | {
         "pruned_html": html_template.format(2),
         "last_action_error": "Hey, this is an error in the past",
     },
-    base_obs | {
+    base_obs
+    | {
         "pruned_html": html_template.format(3),
         "last_action_error": "Hey, there is an error now",
     },
@@ -102,7 +105,7 @@
     ),
     (
         "obs.use_tabs",
-        ("Currently open tabs:","(active tab)"),
+        ("Currently open tabs:", "(active tab)"),
     ),
     (
         "obs.use_focused_element",
@@ -165,7 +168,7 @@ def test_shrinking_observation():
     flags.obs.use_html = True
 
     prompt_maker = MainPrompt(
-        action_set=dp.HighLevelActionSet(),
+        action_set=bgym.HighLevelActionSet(),
         obs_history=OBS_HISTORY,
         actions=ACTIONS,
         memories=MEMORIES,
@@ -231,7 +234,7 @@ def test_main_prompt_elements_present():
     # Initialize MainPrompt
     prompt = str(
         MainPrompt(
-            action_set=dp.HighLevelActionSet(),
+            action_set=bgym.HighLevelActionSet(),
             obs_history=OBS_HISTORY,
             actions=ACTIONS,
             memories=MEMORIES,
@@ -253,4 +256,3 @@ def test_main_prompt_elements_present():
     test_main_prompt_elements_present()
     # for flag, expected_prompts in FLAG_EXPECTED_PROMPT:
     #     test_main_prompt_elements_gone_one_at_a_time(flag, expected_prompts)
-