refactor: rename use_osworld_obs_preprocessor to skip_preprocessing for clarity

Author: amanjaiswal73892

src/agentlab/agents/tool_use_agent/tool_use_agent.py

@@ -166,7 +166,7 @@ class Obs(Block):
     use_tabs: bool = False
     add_mouse_pointer: bool = False
     use_zoomed_webpage: bool = False
-    use_osworld_obs_preprocessor: bool = False
+    skip_preprocessing: bool = False
 
     def apply(
         self, llm, discussion: StructuredDiscussion, obs: dict, last_llm_output: LLMOutput
@@ -382,7 +382,7 @@ def set_benchmark(self, benchmark: AgentLabBenchmark | BgymBenchmark, demo_mode:
         """Set benchmark specific flags."""
         benchmark_name = benchmark.name
         if benchmark_name == "osworld":
-            self.config.obs.use_osworld_obs_preprocessor = True
+            self.config.obs.skip_preprocessing = True
 
 
 class ToolUseAgent(bgym.Agent):
@@ -414,8 +414,8 @@ def __init__(
 
     def obs_preprocessor(self, obs):
         obs = copy(obs)
-        if self.config.obs.use_osworld_obs_preprocessor:
-            return self.osworld_obs_preprocessor(obs)
+        if self.config.obs.skip_preprocessing:
+            return obs
         page = obs.pop("page", None)
         if page is not None:
             obs["screenshot"] = extract_screenshot(page)
@@ -442,10 +442,6 @@ def obs_preprocessor(self, obs):
 
         return obs
 
-    def osworld_obs_preprocessor(self, obs):
-        """Preprocess the observation for OSWorld benchmark."""
-        return obs
-
     def set_task_name(self, task_name: str):
         """Cheater function that is supposed to be called by loop.py before callling get_action"""
         self.task_name = task_name