android-env 1.2.1__py3-none-any.whl → 1.2.3__py3-none-any.whl

android_env/proto/state.proto

@@ -0,0 +1,63 @@
+// Copyright 2024 DeepMind Technologies Limited.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+syntax = "proto3";
+
+package android_env;
+
+option java_multiple_files = true;
+
+message SaveStateRequest {
+  map<string, string> args = 1;
+}
+
+message LoadStateRequest {
+  map<string, string> args = 1;
+}
+
+message SaveStateResponse {
+  enum Status {
+    // Reserved value for unset statuses.
+    UNDEFINED = 0;
+    // Returned when everything goes well.
+    OK = 1;
+    // Returned when something internal did not work as expected.
+    ERROR = 2;
+  }
+  Status status = 1;
+  // `error_message` is only populated in case of errors.
+  string error_message = 2;
+
+  // Any additional info returned during the request; e.g., file paths or sizes.
+  map<string, string> additional_info = 3;
+}
+
+message LoadStateResponse {
+  enum Status {
+    // Reserved value for unset statuses.
+    UNDEFINED = 0;
+    // Returned when everything goes well.
+    OK = 1;
+    // Returned when there is no state to load.
+    NOT_FOUND = 2;
+    // Returned when something internal did not work as expected.
+    ERROR = 3;
+  }
+  Status status = 1;
+  // `error_message` is only populated in case of errors.
+  string error_message = 2;
+
+  // Any additional info returned during the request; e.g., file paths or sizes.
+  map<string, string> additional_info = 3;
+}

android_env/proto/state_pb2.py

@@ -0,0 +1,63 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: android_env/proto/state.proto
+# Protobuf Python Version: 5.29.0
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'android_env/proto/state.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1d\x61ndroid_env/proto/state.proto\x12\x0b\x61ndroid_env\"v\n\x10SaveStateRequest\x12\x35\n\x04\x61rgs\x18\x01 \x03(\x0b\x32\'.android_env.SaveStateRequest.ArgsEntry\x1a+\n\tArgsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"v\n\x10LoadStateRequest\x12\x35\n\x04\x61rgs\x18\x01 \x03(\x0b\x32\'.android_env.LoadStateRequest.ArgsEntry\x1a+\n\tArgsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x91\x02\n\x11SaveStateResponse\x12\x35\n\x06status\x18\x01 \x01(\x0e\x32%.android_env.SaveStateResponse.Status\x12\x15\n\rerror_message\x18\x02 \x01(\t\x12K\n\x0f\x61\x64\x64itional_info\x18\x03 \x03(\x0b\x32\x32.android_env.SaveStateResponse.AdditionalInfoEntry\x1a\x35\n\x13\x41\x64\x64itionalInfoEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"*\n\x06Status\x12\r\n\tUNDEFINED\x10\x00\x12\x06\n\x02OK\x10\x01\x12\t\n\x05\x45RROR\x10\x02\"\xa0\x02\n\x11LoadStateResponse\x12\x35\n\x06status\x18\x01 \x01(\x0e\x32%.android_env.LoadStateResponse.Status\x12\x15\n\rerror_message\x18\x02 \x01(\t\x12K\n\x0f\x61\x64\x64itional_info\x18\x03 \x03(\x0b\x32\x32.android_env.LoadStateResponse.AdditionalInfoEntry\x1a\x35\n\x13\x41\x64\x64itionalInfoEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"9\n\x06Status\x12\r\n\tUNDEFINED\x10\x00\x12\x06\n\x02OK\x10\x01\x12\r\n\tNOT_FOUND\x10\x02\x12\t\n\x05\x45RROR\x10\x03\x42\x02P\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'android_env.proto.state_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'P\001'
+  _globals['_SAVESTATEREQUEST_ARGSENTRY']._loaded_options = None
+  _globals['_SAVESTATEREQUEST_ARGSENTRY']._serialized_options = b'8\001'
+  _globals['_LOADSTATEREQUEST_ARGSENTRY']._loaded_options = None
+  _globals['_LOADSTATEREQUEST_ARGSENTRY']._serialized_options = b'8\001'
+  _globals['_SAVESTATERESPONSE_ADDITIONALINFOENTRY']._loaded_options = None
+  _globals['_SAVESTATERESPONSE_ADDITIONALINFOENTRY']._serialized_options = b'8\001'
+  _globals['_LOADSTATERESPONSE_ADDITIONALINFOENTRY']._loaded_options = None
+  _globals['_LOADSTATERESPONSE_ADDITIONALINFOENTRY']._serialized_options = b'8\001'
+  _globals['_SAVESTATEREQUEST']._serialized_start=46
+  _globals['_SAVESTATEREQUEST']._serialized_end=164
+  _globals['_SAVESTATEREQUEST_ARGSENTRY']._serialized_start=121
+  _globals['_SAVESTATEREQUEST_ARGSENTRY']._serialized_end=164
+  _globals['_LOADSTATEREQUEST']._serialized_start=166
+  _globals['_LOADSTATEREQUEST']._serialized_end=284
+  _globals['_LOADSTATEREQUEST_ARGSENTRY']._serialized_start=121
+  _globals['_LOADSTATEREQUEST_ARGSENTRY']._serialized_end=164
+  _globals['_SAVESTATERESPONSE']._serialized_start=287
+  _globals['_SAVESTATERESPONSE']._serialized_end=560
+  _globals['_SAVESTATERESPONSE_ADDITIONALINFOENTRY']._serialized_start=463
+  _globals['_SAVESTATERESPONSE_ADDITIONALINFOENTRY']._serialized_end=516
+  _globals['_SAVESTATERESPONSE_STATUS']._serialized_start=518
+  _globals['_SAVESTATERESPONSE_STATUS']._serialized_end=560
+  _globals['_LOADSTATERESPONSE']._serialized_start=563
+  _globals['_LOADSTATERESPONSE']._serialized_end=851
+  _globals['_LOADSTATERESPONSE_ADDITIONALINFOENTRY']._serialized_start=463
+  _globals['_LOADSTATERESPONSE_ADDITIONALINFOENTRY']._serialized_end=516
+  _globals['_LOADSTATERESPONSE_STATUS']._serialized_start=794
+  _globals['_LOADSTATERESPONSE_STATUS']._serialized_end=851
+# @@protoc_insertion_point(module_scope)

android_env/proto/state_pb2.pyi

@@ -0,0 +1,85 @@
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class SaveStateRequest(_message.Message):
+    __slots__ = ("args",)
+    class ArgsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    ARGS_FIELD_NUMBER: _ClassVar[int]
+    args: _containers.ScalarMap[str, str]
+    def __init__(self, args: _Optional[_Mapping[str, str]] = ...) -> None: ...
+
+class LoadStateRequest(_message.Message):
+    __slots__ = ("args",)
+    class ArgsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    ARGS_FIELD_NUMBER: _ClassVar[int]
+    args: _containers.ScalarMap[str, str]
+    def __init__(self, args: _Optional[_Mapping[str, str]] = ...) -> None: ...
+
+class SaveStateResponse(_message.Message):
+    __slots__ = ("status", "error_message", "additional_info")
+    class Status(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+        __slots__ = ()
+        UNDEFINED: _ClassVar[SaveStateResponse.Status]
+        OK: _ClassVar[SaveStateResponse.Status]
+        ERROR: _ClassVar[SaveStateResponse.Status]
+    UNDEFINED: SaveStateResponse.Status
+    OK: SaveStateResponse.Status
+    ERROR: SaveStateResponse.Status
+    class AdditionalInfoEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    ERROR_MESSAGE_FIELD_NUMBER: _ClassVar[int]
+    ADDITIONAL_INFO_FIELD_NUMBER: _ClassVar[int]
+    status: SaveStateResponse.Status
+    error_message: str
+    additional_info: _containers.ScalarMap[str, str]
+    def __init__(self, status: _Optional[_Union[SaveStateResponse.Status, str]] = ..., error_message: _Optional[str] = ..., additional_info: _Optional[_Mapping[str, str]] = ...) -> None: ...
+
+class LoadStateResponse(_message.Message):
+    __slots__ = ("status", "error_message", "additional_info")
+    class Status(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+        __slots__ = ()
+        UNDEFINED: _ClassVar[LoadStateResponse.Status]
+        OK: _ClassVar[LoadStateResponse.Status]
+        NOT_FOUND: _ClassVar[LoadStateResponse.Status]
+        ERROR: _ClassVar[LoadStateResponse.Status]
+    UNDEFINED: LoadStateResponse.Status
+    OK: LoadStateResponse.Status
+    NOT_FOUND: LoadStateResponse.Status
+    ERROR: LoadStateResponse.Status
+    class AdditionalInfoEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    ERROR_MESSAGE_FIELD_NUMBER: _ClassVar[int]
+    ADDITIONAL_INFO_FIELD_NUMBER: _ClassVar[int]
+    status: LoadStateResponse.Status
+    error_message: str
+    additional_info: _containers.ScalarMap[str, str]
+    def __init__(self, status: _Optional[_Union[LoadStateResponse.Status, str]] = ..., error_message: _Optional[str] = ..., additional_info: _Optional[_Mapping[str, str]] = ...) -> None: ...

android_env/proto/state_pb2_grpc.py

@@ -0,0 +1,24 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+
+GRPC_GENERATED_VERSION = '1.71.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in android_env/proto/state_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )

android_env/proto/task.proto

@@ -1,4 +1,4 @@
-// Copyright 2022 DeepMind Technologies Limited.
+// Copyright 2024 DeepMind Technologies Limited.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -102,6 +102,7 @@ message ArraySpec {
     STRING_U25 = 14;
     STRING_U250 = 15;
     STRING = 16;  // String without max length
+    OBJECT = 17;
   }
 
   // Data type of elements we expect to see in an array of this spec.
@@ -189,6 +190,9 @@ message Task {
   // Defines parameters for parsing messages from logcat.
   LogParsingConfig log_parsing_config = 9;
 
+  // NOTE: This field is deprecated and will be removed from this Task
+  // definition soon.
+  //
   // (Optional): The task may also define extras to help the RL agent.
   // An Extra in AndroidEnv is any information that apps may send to aid the
   // understanding of the task. The type of information sent through this

android_env/proto/task_pb2.py

@@ -1,11 +1,22 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
 # source: android_env/proto/task.proto
+# Protobuf Python Version: 5.29.0
 """Generated protocol buffer code."""
-from google.protobuf.internal import builder as _builder
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
 from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'android_env/proto/task.proto'
+)
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -14,35 +25,35 @@ _sym_db = _symbol_database.Default()
 from android_env.proto import adb_pb2 as android__env_dot_proto_dot_adb__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1c\x61ndroid_env/proto/task.proto\x12\x0b\x61ndroid_env\x1a\x1b\x61ndroid_env/proto/adb.proto\":\n\tAppScreen\x12\x10\n\x08\x61\x63tivity\x18\x01 \x01(\t\x12\x1b\n\x13view_hierarchy_path\x18\x02 \x03(\t\"S\n\x10WaitForAppScreen\x12*\n\napp_screen\x18\x01 \x01(\x0b\x32\x16.android_env.AppScreen\x12\x13\n\x0btimeout_sec\x18\x02 \x01(\x02\"9\n\x0c\x43heckInstall\x12\x14\n\x0cpackage_name\x18\x01 \x01(\t\x12\x13\n\x0btimeout_sec\x18\x02 \x01(\x02\"\x19\n\x05Sleep\x12\x10\n\x08time_sec\x18\x01 \x01(\x02\"\xa2\x01\n\x10SuccessCondition\x12\x13\n\x0bnum_retries\x18\x01 \x01(\x05\x12<\n\x13wait_for_app_screen\x18\x02 \x01(\x0b\x32\x1d.android_env.WaitForAppScreenH\x00\x12\x32\n\rcheck_install\x18\x03 \x01(\x0b\x32\x19.android_env.CheckInstallH\x00\x42\x07\n\x05\x63heck\"\xa2\x01\n\tSetupStep\x12\x38\n\x11success_condition\x18\x01 \x01(\x0b\x32\x1d.android_env.SuccessCondition\x12.\n\x0b\x61\x64\x62_request\x18\x02 \x01(\x0b\x32\x17.android_env.AdbRequestH\x00\x12#\n\x05sleep\x18\x03 \x01(\x0b\x32\x12.android_env.SleepH\x00\x42\x06\n\x04step\"\xc3\x02\n\tArraySpec\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\r\n\x05shape\x18\x02 \x03(\x05\x12.\n\x05\x64type\x18\x03 \x01(\x0e\x32\x1f.android_env.ArraySpec.DataType\"\xe8\x01\n\x08\x44\x61taType\x12\x15\n\x11INVALID_DATA_TYPE\x10\x00\x12\t\n\x05\x46LOAT\x10\x01\x12\n\n\x06\x44OUBLE\x10\x02\x12\x08\n\x04INT8\x10\x03\x12\t\n\x05INT16\x10\x04\x12\t\n\x05INT32\x10\x05\x12\t\n\x05INT64\x10\x06\x12\t\n\x05UINT8\x10\x07\x12\n\n\x06UINT16\x10\x08\x12\n\n\x06UINT32\x10\t\x12\n\n\x06UINT64\x10\n\x12\x08\n\x04\x42OOL\x10\x0b\x12\r\n\tSTRING_U1\x10\x0c\x12\x0e\n\nSTRING_U16\x10\r\x12\x0e\n\nSTRING_U25\x10\x0e\x12\x0f\n\x0bSTRING_U250\x10\x0f\x12\n\n\x06STRING\x10\x10\"\xc2\x02\n\x10LogParsingConfig\x12\x0f\n\x07\x66ilters\x18\x01 \x03(\t\x12=\n\x0blog_regexps\x18\x02 \x01(\x0b\x32(.android_env.LogParsingConfig.LogRegexps\x1a\xdd\x01\n\nLogRegexps\x12\r\n\x05score\x18\x01 \x01(\t\x12\x0e\n\x06reward\x18\x02 \x03(\t\x12\x13\n\x0b\x65pisode_end\x18\x03 \x03(\t\x12\r\n\x05\x65xtra\x18\x04 \x03(\t\x12\x12\n\njson_extra\x18\x05 \x03(\t\x12J\n\x0creward_event\x18\x06 \x03(\x0b\x32\x34.android_env.LogParsingConfig.LogRegexps.RewardEvent\x1a,\n\x0bRewardEvent\x12\r\n\x05\x65vent\x18\x01 \x01(\t\x12\x0e\n\x06reward\x18\x02 \x01(\x02\"\xe0\x02\n\x04Task\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12+\n\x0bsetup_steps\x18\x04 \x03(\x0b\x32\x16.android_env.SetupStep\x12+\n\x0breset_steps\x18\x05 \x03(\x0b\x32\x16.android_env.SetupStep\x12\x33\n\x13\x65xpected_app_screen\x18\x06 \x01(\x0b\x32\x16.android_env.AppScreen\x12\x17\n\x0fmax_episode_sec\x18\x07 \x01(\x02\x12\x19\n\x11max_episode_steps\x18\x08 \x01(\x05\x12\x39\n\x12log_parsing_config\x18\t \x01(\x0b\x32\x1d.android_env.LogParsingConfig\x12+\n\x0b\x65xtras_spec\x18\n \x03(\x0b\x32\x16.android_env.ArraySpecb\x06proto3')
-
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'android_env.proto.task_pb2', globals())
-if _descriptor._USE_C_DESCRIPTORS == False:
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1c\x61ndroid_env/proto/task.proto\x12\x0b\x61ndroid_env\x1a\x1b\x61ndroid_env/proto/adb.proto\":\n\tAppScreen\x12\x10\n\x08\x61\x63tivity\x18\x01 \x01(\t\x12\x1b\n\x13view_hierarchy_path\x18\x02 \x03(\t\"S\n\x10WaitForAppScreen\x12*\n\napp_screen\x18\x01 \x01(\x0b\x32\x16.android_env.AppScreen\x12\x13\n\x0btimeout_sec\x18\x02 \x01(\x02\"9\n\x0c\x43heckInstall\x12\x14\n\x0cpackage_name\x18\x01 \x01(\t\x12\x13\n\x0btimeout_sec\x18\x02 \x01(\x02\"\x19\n\x05Sleep\x12\x10\n\x08time_sec\x18\x01 \x01(\x02\"\xa2\x01\n\x10SuccessCondition\x12\x13\n\x0bnum_retries\x18\x01 \x01(\x05\x12<\n\x13wait_for_app_screen\x18\x02 \x01(\x0b\x32\x1d.android_env.WaitForAppScreenH\x00\x12\x32\n\rcheck_install\x18\x03 \x01(\x0b\x32\x19.android_env.CheckInstallH\x00\x42\x07\n\x05\x63heck\"\xa2\x01\n\tSetupStep\x12\x38\n\x11success_condition\x18\x01 \x01(\x0b\x32\x1d.android_env.SuccessCondition\x12.\n\x0b\x61\x64\x62_request\x18\x02 \x01(\x0b\x32\x17.android_env.AdbRequestH\x00\x12#\n\x05sleep\x18\x03 \x01(\x0b\x32\x12.android_env.SleepH\x00\x42\x06\n\x04step\"\xcf\x02\n\tArraySpec\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\r\n\x05shape\x18\x02 \x03(\x05\x12.\n\x05\x64type\x18\x03 \x01(\x0e\x32\x1f.android_env.ArraySpec.DataType\"\xf4\x01\n\x08\x44\x61taType\x12\x15\n\x11INVALID_DATA_TYPE\x10\x00\x12\t\n\x05\x46LOAT\x10\x01\x12\n\n\x06\x44OUBLE\x10\x02\x12\x08\n\x04INT8\x10\x03\x12\t\n\x05INT16\x10\x04\x12\t\n\x05INT32\x10\x05\x12\t\n\x05INT64\x10\x06\x12\t\n\x05UINT8\x10\x07\x12\n\n\x06UINT16\x10\x08\x12\n\n\x06UINT32\x10\t\x12\n\n\x06UINT64\x10\n\x12\x08\n\x04\x42OOL\x10\x0b\x12\r\n\tSTRING_U1\x10\x0c\x12\x0e\n\nSTRING_U16\x10\r\x12\x0e\n\nSTRING_U25\x10\x0e\x12\x0f\n\x0bSTRING_U250\x10\x0f\x12\n\n\x06STRING\x10\x10\x12\n\n\x06OBJECT\x10\x11\"\xc2\x02\n\x10LogParsingConfig\x12\x0f\n\x07\x66ilters\x18\x01 \x03(\t\x12=\n\x0blog_regexps\x18\x02 \x01(\x0b\x32(.android_env.LogParsingConfig.LogRegexps\x1a\xdd\x01\n\nLogRegexps\x12\r\n\x05score\x18\x01 \x01(\t\x12\x0e\n\x06reward\x18\x02 \x03(\t\x12\x13\n\x0b\x65pisode_end\x18\x03 \x03(\t\x12\r\n\x05\x65xtra\x18\x04 \x03(\t\x12\x12\n\njson_extra\x18\x05 \x03(\t\x12J\n\x0creward_event\x18\x06 \x03(\x0b\x32\x34.android_env.LogParsingConfig.LogRegexps.RewardEvent\x1a,\n\x0bRewardEvent\x12\r\n\x05\x65vent\x18\x01 \x01(\t\x12\x0e\n\x06reward\x18\x02 \x01(\x02\"\xe0\x02\n\x04Task\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12+\n\x0bsetup_steps\x18\x04 \x03(\x0b\x32\x16.android_env.SetupStep\x12+\n\x0breset_steps\x18\x05 \x03(\x0b\x32\x16.android_env.SetupStep\x12\x33\n\x13\x65xpected_app_screen\x18\x06 \x01(\x0b\x32\x16.android_env.AppScreen\x12\x17\n\x0fmax_episode_sec\x18\x07 \x01(\x02\x12\x19\n\x11max_episode_steps\x18\x08 \x01(\x05\x12\x39\n\x12log_parsing_config\x18\t \x01(\x0b\x32\x1d.android_env.LogParsingConfig\x12+\n\x0b\x65xtras_spec\x18\n \x03(\x0b\x32\x16.android_env.ArraySpecb\x06proto3')
 
-  DESCRIPTOR._options = None
-  _APPSCREEN._serialized_start=74
-  _APPSCREEN._serialized_end=132
-  _WAITFORAPPSCREEN._serialized_start=134
-  _WAITFORAPPSCREEN._serialized_end=217
-  _CHECKINSTALL._serialized_start=219
-  _CHECKINSTALL._serialized_end=276
-  _SLEEP._serialized_start=278
-  _SLEEP._serialized_end=303
-  _SUCCESSCONDITION._serialized_start=306
-  _SUCCESSCONDITION._serialized_end=468
-  _SETUPSTEP._serialized_start=471
-  _SETUPSTEP._serialized_end=633
-  _ARRAYSPEC._serialized_start=636
-  _ARRAYSPEC._serialized_end=959
-  _ARRAYSPEC_DATATYPE._serialized_start=727
-  _ARRAYSPEC_DATATYPE._serialized_end=959
-  _LOGPARSINGCONFIG._serialized_start=962
-  _LOGPARSINGCONFIG._serialized_end=1284
-  _LOGPARSINGCONFIG_LOGREGEXPS._serialized_start=1063
-  _LOGPARSINGCONFIG_LOGREGEXPS._serialized_end=1284
-  _LOGPARSINGCONFIG_LOGREGEXPS_REWARDEVENT._serialized_start=1240
-  _LOGPARSINGCONFIG_LOGREGEXPS_REWARDEVENT._serialized_end=1284
-  _TASK._serialized_start=1287
-  _TASK._serialized_end=1639
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'android_env.proto.task_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_APPSCREEN']._serialized_start=74
+  _globals['_APPSCREEN']._serialized_end=132
+  _globals['_WAITFORAPPSCREEN']._serialized_start=134
+  _globals['_WAITFORAPPSCREEN']._serialized_end=217
+  _globals['_CHECKINSTALL']._serialized_start=219
+  _globals['_CHECKINSTALL']._serialized_end=276
+  _globals['_SLEEP']._serialized_start=278
+  _globals['_SLEEP']._serialized_end=303
+  _globals['_SUCCESSCONDITION']._serialized_start=306
+  _globals['_SUCCESSCONDITION']._serialized_end=468
+  _globals['_SETUPSTEP']._serialized_start=471
+  _globals['_SETUPSTEP']._serialized_end=633
+  _globals['_ARRAYSPEC']._serialized_start=636
+  _globals['_ARRAYSPEC']._serialized_end=971
+  _globals['_ARRAYSPEC_DATATYPE']._serialized_start=727
+  _globals['_ARRAYSPEC_DATATYPE']._serialized_end=971
+  _globals['_LOGPARSINGCONFIG']._serialized_start=974
+  _globals['_LOGPARSINGCONFIG']._serialized_end=1296
+  _globals['_LOGPARSINGCONFIG_LOGREGEXPS']._serialized_start=1075
+  _globals['_LOGPARSINGCONFIG_LOGREGEXPS']._serialized_end=1296
+  _globals['_LOGPARSINGCONFIG_LOGREGEXPS_REWARDEVENT']._serialized_start=1252
+  _globals['_LOGPARSINGCONFIG_LOGREGEXPS_REWARDEVENT']._serialized_end=1296
+  _globals['_TASK']._serialized_start=1299
+  _globals['_TASK']._serialized_end=1651
 # @@protoc_insertion_point(module_scope)

android_env/proto/task_pb2.pyi

@@ -0,0 +1,160 @@
+from android_env.proto import adb_pb2 as _adb_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class AppScreen(_message.Message):
+    __slots__ = ("activity", "view_hierarchy_path")
+    ACTIVITY_FIELD_NUMBER: _ClassVar[int]
+    VIEW_HIERARCHY_PATH_FIELD_NUMBER: _ClassVar[int]
+    activity: str
+    view_hierarchy_path: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, activity: _Optional[str] = ..., view_hierarchy_path: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class WaitForAppScreen(_message.Message):
+    __slots__ = ("app_screen", "timeout_sec")
+    APP_SCREEN_FIELD_NUMBER: _ClassVar[int]
+    TIMEOUT_SEC_FIELD_NUMBER: _ClassVar[int]
+    app_screen: AppScreen
+    timeout_sec: float
+    def __init__(self, app_screen: _Optional[_Union[AppScreen, _Mapping]] = ..., timeout_sec: _Optional[float] = ...) -> None: ...
+
+class CheckInstall(_message.Message):
+    __slots__ = ("package_name", "timeout_sec")
+    PACKAGE_NAME_FIELD_NUMBER: _ClassVar[int]
+    TIMEOUT_SEC_FIELD_NUMBER: _ClassVar[int]
+    package_name: str
+    timeout_sec: float
+    def __init__(self, package_name: _Optional[str] = ..., timeout_sec: _Optional[float] = ...) -> None: ...
+
+class Sleep(_message.Message):
+    __slots__ = ("time_sec",)
+    TIME_SEC_FIELD_NUMBER: _ClassVar[int]
+    time_sec: float
+    def __init__(self, time_sec: _Optional[float] = ...) -> None: ...
+
+class SuccessCondition(_message.Message):
+    __slots__ = ("num_retries", "wait_for_app_screen", "check_install")
+    NUM_RETRIES_FIELD_NUMBER: _ClassVar[int]
+    WAIT_FOR_APP_SCREEN_FIELD_NUMBER: _ClassVar[int]
+    CHECK_INSTALL_FIELD_NUMBER: _ClassVar[int]
+    num_retries: int
+    wait_for_app_screen: WaitForAppScreen
+    check_install: CheckInstall
+    def __init__(self, num_retries: _Optional[int] = ..., wait_for_app_screen: _Optional[_Union[WaitForAppScreen, _Mapping]] = ..., check_install: _Optional[_Union[CheckInstall, _Mapping]] = ...) -> None: ...
+
+class SetupStep(_message.Message):
+    __slots__ = ("success_condition", "adb_request", "sleep")
+    SUCCESS_CONDITION_FIELD_NUMBER: _ClassVar[int]
+    ADB_REQUEST_FIELD_NUMBER: _ClassVar[int]
+    SLEEP_FIELD_NUMBER: _ClassVar[int]
+    success_condition: SuccessCondition
+    adb_request: _adb_pb2.AdbRequest
+    sleep: Sleep
+    def __init__(self, success_condition: _Optional[_Union[SuccessCondition, _Mapping]] = ..., adb_request: _Optional[_Union[_adb_pb2.AdbRequest, _Mapping]] = ..., sleep: _Optional[_Union[Sleep, _Mapping]] = ...) -> None: ...
+
+class ArraySpec(_message.Message):
+    __slots__ = ("name", "shape", "dtype")
+    class DataType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+        __slots__ = ()
+        INVALID_DATA_TYPE: _ClassVar[ArraySpec.DataType]
+        FLOAT: _ClassVar[ArraySpec.DataType]
+        DOUBLE: _ClassVar[ArraySpec.DataType]
+        INT8: _ClassVar[ArraySpec.DataType]
+        INT16: _ClassVar[ArraySpec.DataType]
+        INT32: _ClassVar[ArraySpec.DataType]
+        INT64: _ClassVar[ArraySpec.DataType]
+        UINT8: _ClassVar[ArraySpec.DataType]
+        UINT16: _ClassVar[ArraySpec.DataType]
+        UINT32: _ClassVar[ArraySpec.DataType]
+        UINT64: _ClassVar[ArraySpec.DataType]
+        BOOL: _ClassVar[ArraySpec.DataType]
+        STRING_U1: _ClassVar[ArraySpec.DataType]
+        STRING_U16: _ClassVar[ArraySpec.DataType]
+        STRING_U25: _ClassVar[ArraySpec.DataType]
+        STRING_U250: _ClassVar[ArraySpec.DataType]
+        STRING: _ClassVar[ArraySpec.DataType]
+        OBJECT: _ClassVar[ArraySpec.DataType]
+    INVALID_DATA_TYPE: ArraySpec.DataType
+    FLOAT: ArraySpec.DataType
+    DOUBLE: ArraySpec.DataType
+    INT8: ArraySpec.DataType
+    INT16: ArraySpec.DataType
+    INT32: ArraySpec.DataType
+    INT64: ArraySpec.DataType
+    UINT8: ArraySpec.DataType
+    UINT16: ArraySpec.DataType
+    UINT32: ArraySpec.DataType
+    UINT64: ArraySpec.DataType
+    BOOL: ArraySpec.DataType
+    STRING_U1: ArraySpec.DataType
+    STRING_U16: ArraySpec.DataType
+    STRING_U25: ArraySpec.DataType
+    STRING_U250: ArraySpec.DataType
+    STRING: ArraySpec.DataType
+    OBJECT: ArraySpec.DataType
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    SHAPE_FIELD_NUMBER: _ClassVar[int]
+    DTYPE_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    shape: _containers.RepeatedScalarFieldContainer[int]
+    dtype: ArraySpec.DataType
+    def __init__(self, name: _Optional[str] = ..., shape: _Optional[_Iterable[int]] = ..., dtype: _Optional[_Union[ArraySpec.DataType, str]] = ...) -> None: ...
+
+class LogParsingConfig(_message.Message):
+    __slots__ = ("filters", "log_regexps")
+    class LogRegexps(_message.Message):
+        __slots__ = ("score", "reward", "episode_end", "extra", "json_extra", "reward_event")
+        class RewardEvent(_message.Message):
+            __slots__ = ("event", "reward")
+            EVENT_FIELD_NUMBER: _ClassVar[int]
+            REWARD_FIELD_NUMBER: _ClassVar[int]
+            event: str
+            reward: float
+            def __init__(self, event: _Optional[str] = ..., reward: _Optional[float] = ...) -> None: ...
+        SCORE_FIELD_NUMBER: _ClassVar[int]
+        REWARD_FIELD_NUMBER: _ClassVar[int]
+        EPISODE_END_FIELD_NUMBER: _ClassVar[int]
+        EXTRA_FIELD_NUMBER: _ClassVar[int]
+        JSON_EXTRA_FIELD_NUMBER: _ClassVar[int]
+        REWARD_EVENT_FIELD_NUMBER: _ClassVar[int]
+        score: str
+        reward: _containers.RepeatedScalarFieldContainer[str]
+        episode_end: _containers.RepeatedScalarFieldContainer[str]
+        extra: _containers.RepeatedScalarFieldContainer[str]
+        json_extra: _containers.RepeatedScalarFieldContainer[str]
+        reward_event: _containers.RepeatedCompositeFieldContainer[LogParsingConfig.LogRegexps.RewardEvent]
+        def __init__(self, score: _Optional[str] = ..., reward: _Optional[_Iterable[str]] = ..., episode_end: _Optional[_Iterable[str]] = ..., extra: _Optional[_Iterable[str]] = ..., json_extra: _Optional[_Iterable[str]] = ..., reward_event: _Optional[_Iterable[_Union[LogParsingConfig.LogRegexps.RewardEvent, _Mapping]]] = ...) -> None: ...
+    FILTERS_FIELD_NUMBER: _ClassVar[int]
+    LOG_REGEXPS_FIELD_NUMBER: _ClassVar[int]
+    filters: _containers.RepeatedScalarFieldContainer[str]
+    log_regexps: LogParsingConfig.LogRegexps
+    def __init__(self, filters: _Optional[_Iterable[str]] = ..., log_regexps: _Optional[_Union[LogParsingConfig.LogRegexps, _Mapping]] = ...) -> None: ...
+
+class Task(_message.Message):
+    __slots__ = ("id", "name", "description", "setup_steps", "reset_steps", "expected_app_screen", "max_episode_sec", "max_episode_steps", "log_parsing_config", "extras_spec")
+    ID_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
+    SETUP_STEPS_FIELD_NUMBER: _ClassVar[int]
+    RESET_STEPS_FIELD_NUMBER: _ClassVar[int]
+    EXPECTED_APP_SCREEN_FIELD_NUMBER: _ClassVar[int]
+    MAX_EPISODE_SEC_FIELD_NUMBER: _ClassVar[int]
+    MAX_EPISODE_STEPS_FIELD_NUMBER: _ClassVar[int]
+    LOG_PARSING_CONFIG_FIELD_NUMBER: _ClassVar[int]
+    EXTRAS_SPEC_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    name: str
+    description: str
+    setup_steps: _containers.RepeatedCompositeFieldContainer[SetupStep]
+    reset_steps: _containers.RepeatedCompositeFieldContainer[SetupStep]
+    expected_app_screen: AppScreen
+    max_episode_sec: float
+    max_episode_steps: int
+    log_parsing_config: LogParsingConfig
+    extras_spec: _containers.RepeatedCompositeFieldContainer[ArraySpec]
+    def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., description: _Optional[str] = ..., setup_steps: _Optional[_Iterable[_Union[SetupStep, _Mapping]]] = ..., reset_steps: _Optional[_Iterable[_Union[SetupStep, _Mapping]]] = ..., expected_app_screen: _Optional[_Union[AppScreen, _Mapping]] = ..., max_episode_sec: _Optional[float] = ..., max_episode_steps: _Optional[int] = ..., log_parsing_config: _Optional[_Union[LogParsingConfig, _Mapping]] = ..., extras_spec: _Optional[_Iterable[_Union[ArraySpec, _Mapping]]] = ...) -> None: ...

android_env/proto/task_pb2_grpc.py

@@ -1,4 +1,24 @@
 # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
 """Client and server classes corresponding to protobuf-defined services."""
 import grpc
+import warnings
 
+
+GRPC_GENERATED_VERSION = '1.71.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in android_env/proto/task_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )

android_env/wrappers/__init__.py

@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 DeepMind Technologies Limited.
+# Copyright 2024 DeepMind Technologies Limited.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.