add convert_obs to env reset function

amanjaiswal73892 · amanjaiswal73892 · commit 3728a3b9133e · 2025-06-27T11:20:22.000-04:00
diff --git a/src/agentlab/benchmarks/osworld.py b/src/agentlab/benchmarks/osworld.py
@@ -62,7 +62,8 @@ def __init__(
         )
 
     def reset(self, seed: int | None = None) -> tuple[dict[str, Any], dict[str, Any]]:
-        obs = self.env.reset(task_config=self.task, seed=seed)
+        raw_obs = self.env.reset(task_config=self.task, seed=seed)
+        obs = self.convert_observation(raw_obs)
         return obs, self.env_info
 
     def step(self, action: str):

Original file line number	Diff line number	Diff line change
`@@ -62,7 +62,8 @@ def __init__(`
`62`	`62`	`)`
`63`	`63`
`64`	`64`	`def reset(self, seed: int \| None = None) -> tuple[dict[str, Any], dict[str, Any]]:`
`65`		`- obs = self.env.reset(task_config=self.task, seed=seed)`
	`65`	`+ raw_obs = self.env.reset(task_config=self.task, seed=seed)`
	`66`	`+ obs = self.convert_observation(raw_obs)`
`66`	`67`	`return obs, self.env_info`
`67`	`68`
`68`	`69`	`def step(self, action: str):`