temp-commit

Author: amanjaiswal73892

main_workarena_debug.py

@@ -42,15 +42,16 @@
 
 # ## select the benchmark to run on
 # benchmark = "miniwob_tiny_test"
-benchmark = "workarena_l1"
+# benchmark = "workarena_l1"
 
 
-benchmark = bgym.DEFAULT_BENCHMARKS[benchmark](n_repeats=4)  # type: bgym.Benchmark
-benchmark = benchmark.subset_from_glob("task_name", "*create*")
+benchmark = bgym.DEFAULT_BENCHMARKS['miniwob']()  # type: bgym.Benchmark
+# benchmark = benchmark.subset_from_glob("task_name", "*create*")
 
-# for env_args in benchmark.env_args_list:
-#     print(env_args.task_name)
-#     env_args.max_steps = 15
+for env_args in benchmark.env_args_list:
+    print(env_args.task_name)
+    env_args.max_steps = 15
+    env_args.seed = 15
 
 relaunch = False
 

src/agentlab/analyze/agent_xray.py

@@ -103,17 +103,10 @@ def update_exp_result(self, episode_id: EpisodeId):
         # Prefer selecting by explicit row index if available
         if episode_id.row_index is not None:
             tmp_df = self.result_df.reset_index(inplace=False)
-            tmp_df["_row_index"] = tmp_df.index
-            sub_df = tmp_df[tmp_df["_row_index"] == episode_id.row_index]
-            if len(sub_df) == 0:
+            if not (0 <= episode_id.row_index < len(tmp_df)):
                 self.exp_result = None
-                raise ValueError(f"Could not find episode for row_index: {episode_id.row_index}")
-            if len(sub_df) > 1:
-                warning(
-                    f"Found multiple rows for row_index: {episode_id.row_index}. Using the first one."
-                )
-            exp_dir = sub_df.iloc[0]["exp_dir"]
-            print(exp_dir)
+                raise ValueError(f"row_index out of range: {episode_id.row_index}")
+            exp_dir = tmp_df.iloc[episode_id.row_index]["exp_dir"]
             self.exp_result = ExpResult(exp_dir)
             self.step = 0
             return
@@ -1031,7 +1024,7 @@ def extract_columns(row: pd.Series):
 
     seed_df = result_df.apply(extract_columns, axis=1)
     # Ensure column order and readability
-    seed_df = seed_df[["seed", "reward", "err", "n_steps","index"]]
+    seed_df = seed_df[["seed", "reward", "err", "n_steps", "index"]]
     return seed_df