From be8bb13eca5463707787b4aa210f3a55b1fcff10 Mon Sep 17 00:00:00 2001
From: DeepMind <copybara-servicebot@google.com>
Date: Thu, 26 Jun 2025 13:11:16 -0700
Subject: [PATCH] Not relevant for this CL, only for the diffbase

PiperOrigin-RevId: 776253231
---
 android_env/components/config_classes.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/android_env/components/config_classes.py b/android_env/components/config_classes.py
index 6b420db5..15d6f0ad 100644
--- a/android_env/components/config_classes.py
+++ b/android_env/components/config_classes.py
@@ -16,6 +16,7 @@
 """Dataclass definitions used for instantiating AndroidEnv components."""
 
 import dataclasses
+import enum
 
 
 @dataclasses.dataclass
@@ -79,6 +80,15 @@ class SimulatorConfig:
   interaction_rate_sec: float = 0.0
 
 
+@enum.unique
+class GPUMode(enum.Enum):
+  """Emulator GPU Mode."""
+
+  HOST = 'host'
+  SWANGLE_INDIRECT = 'swangle_indirect'
+  SWIFTSHADER_INDIRECT = 'swiftshader_indirect'
+
+
 @dataclasses.dataclass
 class EmulatorLauncherConfig:
   """Config class for EmulatorLauncher."""
@@ -104,7 +114,7 @@ class EmulatorLauncherConfig:
   # GPU mode override.
   # Please see
   # https://developer.android.com/studio/run/emulator-acceleration#accel-graphics.
-  gpu_mode: str = 'swangle_indirect'  # Alternative: swiftshader_indirect, host
+  gpu_mode: str = GPUMode.SWANGLE_INDIRECT.value
   # Whether to run in headless mode (i.e. without a graphical window).
   run_headless: bool = True
   # Whether to restrict network access.