Adding a function to install required apps if necessary. A caller can disable the initial installation and postpone it until required. This speeds up the evaluation on a subset of tasks.

PiperOrigin-RevId: 792172207

Author: The android_world Authors

android_world/env/setup_device/apps.py

@@ -75,6 +75,12 @@ class AppSetup(abc.ABC):
   # The short name of the app, as used by adb_utils.
   app_name = ""
 
+  @classmethod
+  def package_name(cls) -> str:
+    return adb_utils.extract_package_name(
+        adb_utils.get_adb_activity(cls.app_name)
+    )
+
   @classmethod
   def setup(cls, env: interface.AsyncEnv) -> None:
     """Performs setup tasks specific to the app."""

android_world/env/setup_device/setup.py

@@ -63,9 +63,24 @@
 )
 
 
-def get_app_mapping(app_name: str) -> Type[apps.AppSetup]:
+def get_installed_packages(env: interface.AsyncEnv) -> frozenset[str]:
+  """Returns the set of installed packages."""
+  return frozenset(adb_utils.get_all_package_names(env.controller.env))
+
+
+def is_package_installed(package_name: str, env: interface.AsyncEnv) -> bool:
+  """Checks if a package is installed."""
+  installed_packages = get_installed_packages(env)
+  return package_name in installed_packages
+
+
+def get_app_mapping(app_name: str) -> Type[apps.AppSetup] | None:
+  if not app_name:
+    return None
   mapping = {app.app_name: app for app in _APPS}
-  return mapping[app_name]
+  if app_name in mapping:
+    return mapping[app_name]
+  return None
 
 
 def get_app_list_to_setup(
@@ -116,6 +131,12 @@ def setup_app(app: Type[apps.AppSetup], env: interface.AsyncEnv) -> None:
   app_snapshot.save_snapshot(app.app_name, env.controller)
 
 
+def install_app_if_not_installed(app_name: str, env: interface.AsyncEnv):
+  """Installs the apk of an app only if the apk is not installed."""
+  path = apps.download_app_data(apk)
+  adb_utils.install_apk(path, raw_env)
+
+
 def maybe_install_app(
     app: Type[apps.AppSetup], env: interface.AsyncEnv
 ) -> None:

android_world/task_evals/task_eval.py

@@ -17,10 +17,12 @@
 import abc
 import random
 from typing import Any
+
 from absl import logging
 from android_world.env import adb_utils
 from android_world.env import device_constants
 from android_world.env import interface
+from android_world.env.setup_device import setup
 from android_world.utils import app_snapshot
 from android_world.utils import datetime_utils
 
@@ -122,6 +124,10 @@ def _initialize_apps(self, env: interface.AsyncEnv) -> None:
         except RuntimeError as error:
           logging.warning("Skipping app snapshot loading : %s", error)
 
+  def install_apps_if_not_installed(self, env: interface.AsyncEnv) -> None:
+    for app_name in self.app_names:
+      setup.install_app_if_not_installed(app_name, env)
+
   @classmethod
   def set_device_time(cls, env: interface.AsyncEnv) -> None:
     """Sets the device time."""