Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add real device simulator to AndroidEnv loader. #241

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions android_env/loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@
from android_env.components import task_manager as task_manager_lib
from android_env.components.simulators.emulator import emulator_simulator
from android_env.components.simulators.fake import fake_simulator
from android_env.components.simulators.real_device import real_device_simulator
from android_env.proto import task_pb2

from google.protobuf import text_format
Expand Down Expand Up @@ -56,6 +57,10 @@ def load(config: config_classes.AndroidEnvConfig) -> environment.AndroidEnv:
simulator = emulator_simulator.EmulatorSimulator(config=config.simulator)
case config_classes.FakeSimulatorConfig():
simulator = fake_simulator.FakeSimulator(config=config.simulator)
case config_classes.RealDeviceConfig():
simulator = real_device_simulator.RealDeviceSimulator(
config=config.simulator
)
case _:
raise ValueError('Unsupported simulator config: {config.simulator}')

Expand Down
Loading