From d3c62f00d45f21a79f095f5600f03fcb4154e9ce Mon Sep 17 00:00:00 2001
From: NingLi <2212363918@qq.com>
Date: Sat, 25 Oct 2025 13:44:11 +0800
Subject: [PATCH] Fix Chrome app setup and Browser task initialization

---
 android_world/env/setup_device/apps.py     | 11 +++++++++--
 android_world/task_evals/single/browser.py |  8 ++++----
 2 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/android_world/env/setup_device/apps.py b/android_world/env/setup_device/apps.py
index bd63f5b13..de35bb620 100644
--- a/android_world/env/setup_device/apps.py
+++ b/android_world/env/setup_device/apps.py
@@ -164,8 +164,15 @@ def setup(cls, env: interface.AsyncEnv) -> None:
     try:
       controller = tools.AndroidToolController(env=env.controller)
       time.sleep(2.0)
-      # Welcome screen.
-      controller.click_element("Accept & continue")
+      try:
+        # Welcome screen.
+        controller.click_element("Accept & continue")
+      except ValueError as e:
+        # Use without an account
+        logging.warn("Can't find `Accept & continue`, try to clock `Use without an account`.")
+        time.sleep(10)
+        action = json_action.JSONAction(action_type='click', x=540, y=2096)
+        actuation.execute_adb_action(action, [], (0, 0), env.controller)
       time.sleep(2.0)
       # Turn on sync?
       controller.click_element("No thanks")
diff --git a/android_world/task_evals/single/browser.py b/android_world/task_evals/single/browser.py
index bf6d413fc..61fd6d667 100644
--- a/android_world/task_evals/single/browser.py
+++ b/android_world/task_evals/single/browser.py
@@ -60,10 +60,10 @@ def initialize_task(self, env: interface.AsyncEnv):
         adb_utils.get_adb_activity('chrome')
     )
 
-    adb_utils.clear_app_data(
-        chrome_activity,
-        env.controller,
-    )
+    # adb_utils.clear_app_data(
+    #     chrome_activity,
+    #     env.controller,
+    # )
     adb_utils.grant_permissions(
         chrome_activity,
         'android.permission.POST_NOTIFICATIONS',