From b1ed0a478511115622edf0e5c1590c93f4cdd855 Mon Sep 17 00:00:00 2001
From: lizhanyuan <949777411@qq.com>
Date: Wed, 25 Mar 2026 23:27:21 +0800
Subject: [PATCH] add a11y_tree recording to trajectory output

---
 lib_run_single.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/lib_run_single.py b/lib_run_single.py
index 78051b8..277cc04 100644
--- a/lib_run_single.py
+++ b/lib_run_single.py
@@ -67,7 +67,8 @@ def run_single_example(agent, env, example, max_steps, instruction, args, exampl
                     "reward": reward,
                     "done": done,
                     "info": info,
-                    "screenshot_file": f"step_{step_idx + 1}_{action_timestamp}.png"
+                    "screenshot_file": f"step_{step_idx + 1}_{action_timestamp}.png",
+                    "a11y_tree": obs.get("accessibility_tree") if isinstance(obs, dict) else None
                 }))
                 f.write("\n")
             if done: