Skip to content

Commit 1aee56c

Browse files
surfnerdsurfnerd
committed
Apply protobuf patch.
1 parent 392cbf6 commit 1aee56c

18 files changed

+221
-119
lines changed

‎ml-agents-envs/mlagents_envs/communicator_objects/agent_action_pb2.pyi

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -22,18 +22,24 @@ from typing_extensions import (
2222
)
2323

2424

25+
builtin___bool = bool
26+
builtin___bytes = bytes
27+
builtin___float = float
28+
builtin___int = int
29+
30+
2531
class AgentActionProto(google___protobuf___message___Message):
2632
DESCRIPTOR: google___protobuf___descriptor___Descriptor = ...
27-
vector_actions = ... # type: google___protobuf___internal___containers___RepeatedScalarFieldContainer[float]
28-
value = ... # type: float
33+
vector_actions = ... # type: google___protobuf___internal___containers___RepeatedScalarFieldContainer[builtin___float]
34+
value = ... # type: builtin___float
2935

3036
def __init__(self,
3137
*,
32-
vector_actions : typing___Optional[typing___Iterable[float]] = None,
33-
value : typing___Optional[float] = None,
38+
vector_actions : typing___Optional[typing___Iterable[builtin___float]] = None,
39+
value : typing___Optional[builtin___float] = None,
3440
) -> None: ...
3541
@classmethod
36-
def FromString(cls, s: bytes) -> AgentActionProto: ...
42+
def FromString(cls, s: builtin___bytes) -> AgentActionProto: ...
3743
def MergeFrom(self, other_msg: google___protobuf___message___Message) -> None: ...
3844
def CopyFrom(self, other_msg: google___protobuf___message___Message) -> None: ...
3945
if sys.version_info >= (3,):

‎ml-agents-envs/mlagents_envs/communicator_objects/agent_info_action_pair_pb2.pyi

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,12 @@ from typing_extensions import (
2525
)
2626

2727

28+
builtin___bool = bool
29+
builtin___bytes = bytes
30+
builtin___float = float
31+
builtin___int = int
32+
33+
2834
class AgentInfoActionPairProto(google___protobuf___message___Message):
2935
DESCRIPTOR: google___protobuf___descriptor___Descriptor = ...
3036

@@ -40,12 +46,12 @@ class AgentInfoActionPairProto(google___protobuf___message___Message):
4046
action_info : typing___Optional[mlagents_envs___communicator_objects___agent_action_pb2___AgentActionProto] = None,
4147
) -> None: ...
4248
@classmethod
43-
def FromString(cls, s: bytes) -> AgentInfoActionPairProto: ...
49+
def FromString(cls, s: builtin___bytes) -> AgentInfoActionPairProto: ...
4450
def MergeFrom(self, other_msg: google___protobuf___message___Message) -> None: ...
4551
def CopyFrom(self, other_msg: google___protobuf___message___Message) -> None: ...
4652
if sys.version_info >= (3,):
47-
def HasField(self, field_name: typing_extensions___Literal[u"action_info",u"agent_info"]) -> bool: ...
53+
def HasField(self, field_name: typing_extensions___Literal[u"action_info",u"agent_info"]) -> builtin___bool: ...
4854
def ClearField(self, field_name: typing_extensions___Literal[u"action_info",u"agent_info"]) -> None: ...
4955
else:
50-
def HasField(self, field_name: typing_extensions___Literal[u"action_info",b"action_info",u"agent_info",b"agent_info"]) -> bool: ...
56+
def HasField(self, field_name: typing_extensions___Literal[u"action_info",b"action_info",u"agent_info",b"agent_info"]) -> builtin___bool: ...
5157
def ClearField(self, field_name: typing_extensions___Literal[u"action_info",b"action_info",u"agent_info",b"agent_info"]) -> None: ...

‎ml-agents-envs/mlagents_envs/communicator_objects/agent_info_pb2.pyi

Lines changed: 17 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -27,28 +27,34 @@ from typing_extensions import (
2727
)
2828

2929

30+
builtin___bool = bool
31+
builtin___bytes = bytes
32+
builtin___float = float
33+
builtin___int = int
34+
35+
3036
class AgentInfoProto(google___protobuf___message___Message):
3137
DESCRIPTOR: google___protobuf___descriptor___Descriptor = ...
32-
reward = ... # type: float
33-
done = ... # type: bool
34-
max_step_reached = ... # type: bool
35-
id = ... # type: int
36-
action_mask = ... # type: google___protobuf___internal___containers___RepeatedScalarFieldContainer[bool]
38+
reward = ... # type: builtin___float
39+
done = ... # type: builtin___bool
40+
max_step_reached = ... # type: builtin___bool
41+
id = ... # type: builtin___int
42+
action_mask = ... # type: google___protobuf___internal___containers___RepeatedScalarFieldContainer[builtin___bool]
3743

3844
@property
3945
def observations(self) -> google___protobuf___internal___containers___RepeatedCompositeFieldContainer[mlagents_envs___communicator_objects___observation_pb2___ObservationProto]: ...
4046

4147
def __init__(self,
4248
*,
43-
reward : typing___Optional[float] = None,
44-
done : typing___Optional[bool] = None,
45-
max_step_reached : typing___Optional[bool] = None,
46-
id : typing___Optional[int] = None,
47-
action_mask : typing___Optional[typing___Iterable[bool]] = None,
49+
reward : typing___Optional[builtin___float] = None,
50+
done : typing___Optional[builtin___bool] = None,
51+
max_step_reached : typing___Optional[builtin___bool] = None,
52+
id : typing___Optional[builtin___int] = None,
53+
action_mask : typing___Optional[typing___Iterable[builtin___bool]] = None,
4854
observations : typing___Optional[typing___Iterable[mlagents_envs___communicator_objects___observation_pb2___ObservationProto]] = None,
4955
) -> None: ...
5056
@classmethod
51-
def FromString(cls, s: bytes) -> AgentInfoProto: ...
57+
def FromString(cls, s: builtin___bytes) -> AgentInfoProto: ...
5258
def MergeFrom(self, other_msg: google___protobuf___message___Message) -> None: ...
5359
def CopyFrom(self, other_msg: google___protobuf___message___Message) -> None: ...
5460
if sys.version_info >= (3,):

‎ml-agents-envs/mlagents_envs/communicator_objects/brain_parameters_pb2.pyi

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -27,24 +27,30 @@ from typing_extensions import (
2727
)
2828

2929

30+
builtin___bool = bool
31+
builtin___bytes = bytes
32+
builtin___float = float
33+
builtin___int = int
34+
35+
3036
class BrainParametersProto(google___protobuf___message___Message):
3137
DESCRIPTOR: google___protobuf___descriptor___Descriptor = ...
32-
vector_action_size = ... # type: google___protobuf___internal___containers___RepeatedScalarFieldContainer[int]
38+
vector_action_size = ... # type: google___protobuf___internal___containers___RepeatedScalarFieldContainer[builtin___int]
3339
vector_action_descriptions = ... # type: google___protobuf___internal___containers___RepeatedScalarFieldContainer[typing___Text]
3440
vector_action_space_type = ... # type: mlagents_envs___communicator_objects___space_type_pb2___SpaceTypeProto
3541
brain_name = ... # type: typing___Text
36-
is_training = ... # type: bool
42+
is_training = ... # type: builtin___bool
3743

3844
def __init__(self,
3945
*,
40-
vector_action_size : typing___Optional[typing___Iterable[int]] = None,
46+
vector_action_size : typing___Optional[typing___Iterable[builtin___int]] = None,
4147
vector_action_descriptions : typing___Optional[typing___Iterable[typing___Text]] = None,
4248
vector_action_space_type : typing___Optional[mlagents_envs___communicator_objects___space_type_pb2___SpaceTypeProto] = None,
4349
brain_name : typing___Optional[typing___Text] = None,
44-
is_training : typing___Optional[bool] = None,
50+
is_training : typing___Optional[builtin___bool] = None,
4551
) -> None: ...
4652
@classmethod
47-
def FromString(cls, s: bytes) -> BrainParametersProto: ...
53+
def FromString(cls, s: builtin___bytes) -> BrainParametersProto: ...
4854
def MergeFrom(self, other_msg: google___protobuf___message___Message) -> None: ...
4955
def CopyFrom(self, other_msg: google___protobuf___message___Message) -> None: ...
5056
if sys.version_info >= (3,):

‎ml-agents-envs/mlagents_envs/communicator_objects/command_pb2.pyi

Lines changed: 16 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -15,21 +15,25 @@ from typing import (
1515
)
1616

1717

18-
class CommandProto(int):
18+
builtin___int = int
19+
builtin___str = str
20+
21+
22+
class CommandProto(builtin___int):
1923
DESCRIPTOR: google___protobuf___descriptor___EnumDescriptor = ...
2024
@classmethod
21-
def Name(cls, number: int) -> str: ...
25+
def Name(cls, number: builtin___int) -> builtin___str: ...
2226
@classmethod
23-
def Value(cls, name: str) -> CommandProto: ...
27+
def Value(cls, name: builtin___str) -> 'CommandProto': ...
2428
@classmethod
25-
def keys(cls) -> typing___List[str]: ...
29+
def keys(cls) -> typing___List[builtin___str]: ...
2630
@classmethod
27-
def values(cls) -> typing___List[CommandProto]: ...
31+
def values(cls) -> typing___List['CommandProto']: ...
2832
@classmethod
29-
def items(cls) -> typing___List[typing___Tuple[str, CommandProto]]: ...
30-
STEP = typing___cast(CommandProto, 0)
31-
RESET = typing___cast(CommandProto, 1)
32-
QUIT = typing___cast(CommandProto, 2)
33-
STEP = typing___cast(CommandProto, 0)
34-
RESET = typing___cast(CommandProto, 1)
35-
QUIT = typing___cast(CommandProto, 2)
33+
def items(cls) -> typing___List[typing___Tuple[builtin___str, 'CommandProto']]: ...
34+
STEP = typing___cast('CommandProto', 0)
35+
RESET = typing___cast('CommandProto', 1)
36+
QUIT = typing___cast('CommandProto', 2)
37+
STEP = typing___cast('CommandProto', 0)
38+
RESET = typing___cast('CommandProto', 1)
39+
QUIT = typing___cast('CommandProto', 2)

‎ml-agents-envs/mlagents_envs/communicator_objects/custom_reset_parameters_pb2.pyi

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,12 +9,15 @@ from google.protobuf.message import (
99
)
1010

1111

12+
builtin___bytes = bytes
13+
14+
1215
class CustomResetParametersProto(google___protobuf___message___Message):
1316
DESCRIPTOR: google___protobuf___descriptor___Descriptor = ...
1417

1518
def __init__(self,
1619
) -> None: ...
1720
@classmethod
18-
def FromString(cls, s: bytes) -> CustomResetParametersProto: ...
21+
def FromString(cls, s: builtin___bytes) -> CustomResetParametersProto: ...
1922
def MergeFrom(self, other_msg: google___protobuf___message___Message) -> None: ...
2023
def CopyFrom(self, other_msg: google___protobuf___message___Message) -> None: ...

‎ml-agents-envs/mlagents_envs/communicator_objects/demonstration_meta_pb2.pyi

Lines changed: 15 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -18,24 +18,30 @@ from typing_extensions import (
1818
)
1919

2020

21+
builtin___bool = bool
22+
builtin___bytes = bytes
23+
builtin___float = float
24+
builtin___int = int
25+
26+
2127
class DemonstrationMetaProto(google___protobuf___message___Message):
2228
DESCRIPTOR: google___protobuf___descriptor___Descriptor = ...
23-
api_version = ... # type: int
29+
api_version = ... # type: builtin___int
2430
demonstration_name = ... # type: typing___Text
25-
number_steps = ... # type: int
26-
number_episodes = ... # type: int
27-
mean_reward = ... # type: float
31+
number_steps = ... # type: builtin___int
32+
number_episodes = ... # type: builtin___int
33+
mean_reward = ... # type: builtin___float
2834

2935
def __init__(self,
3036
*,
31-
api_version : typing___Optional[int] = None,
37+
api_version : typing___Optional[builtin___int] = None,
3238
demonstration_name : typing___Optional[typing___Text] = None,
33-
number_steps : typing___Optional[int] = None,
34-
number_episodes : typing___Optional[int] = None,
35-
mean_reward : typing___Optional[float] = None,
39+
number_steps : typing___Optional[builtin___int] = None,
40+
number_episodes : typing___Optional[builtin___int] = None,
41+
mean_reward : typing___Optional[builtin___float] = None,
3642
) -> None: ...
3743
@classmethod
38-
def FromString(cls, s: bytes) -> DemonstrationMetaProto: ...
44+
def FromString(cls, s: builtin___bytes) -> DemonstrationMetaProto: ...
3945
def MergeFrom(self, other_msg: google___protobuf___message___Message) -> None: ...
4046
def CopyFrom(self, other_msg: google___protobuf___message___Message) -> None: ...
4147
if sys.version_info >= (3,):

‎ml-agents-envs/mlagents_envs/communicator_objects/engine_configuration_pb2.pyi

Lines changed: 19 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -17,26 +17,32 @@ from typing_extensions import (
1717
)
1818

1919

20+
builtin___bool = bool
21+
builtin___bytes = bytes
22+
builtin___float = float
23+
builtin___int = int
24+
25+
2026
class EngineConfigurationProto(google___protobuf___message___Message):
2127
DESCRIPTOR: google___protobuf___descriptor___Descriptor = ...
22-
width = ... # type: int
23-
height = ... # type: int
24-
quality_level = ... # type: int
25-
time_scale = ... # type: float
26-
target_frame_rate = ... # type: int
27-
show_monitor = ... # type: bool
28+
width = ... # type: builtin___int
29+
height = ... # type: builtin___int
30+
quality_level = ... # type: builtin___int
31+
time_scale = ... # type: builtin___float
32+
target_frame_rate = ... # type: builtin___int
33+
show_monitor = ... # type: builtin___bool
2834

2935
def __init__(self,
3036
*,
31-
width : typing___Optional[int] = None,
32-
height : typing___Optional[int] = None,
33-
quality_level : typing___Optional[int] = None,
34-
time_scale : typing___Optional[float] = None,
35-
target_frame_rate : typing___Optional[int] = None,
36-
show_monitor : typing___Optional[bool] = None,
37+
width : typing___Optional[builtin___int] = None,
38+
height : typing___Optional[builtin___int] = None,
39+
quality_level : typing___Optional[builtin___int] = None,
40+
time_scale : typing___Optional[builtin___float] = None,
41+
target_frame_rate : typing___Optional[builtin___int] = None,
42+
show_monitor : typing___Optional[builtin___bool] = None,
3743
) -> None: ...
3844
@classmethod
39-
def FromString(cls, s: bytes) -> EngineConfigurationProto: ...
45+
def FromString(cls, s: builtin___bytes) -> EngineConfigurationProto: ...
4046
def MergeFrom(self, other_msg: google___protobuf___message___Message) -> None: ...
4147
def CopyFrom(self, other_msg: google___protobuf___message___Message) -> None: ...
4248
if sys.version_info >= (3,):

‎ml-agents-envs/mlagents_envs/communicator_objects/header_pb2.pyi

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,18 +18,24 @@ from typing_extensions import (
1818
)
1919

2020

21+
builtin___bool = bool
22+
builtin___bytes = bytes
23+
builtin___float = float
24+
builtin___int = int
25+
26+
2127
class HeaderProto(google___protobuf___message___Message):
2228
DESCRIPTOR: google___protobuf___descriptor___Descriptor = ...
23-
status = ... # type: int
29+
status = ... # type: builtin___int
2430
message = ... # type: typing___Text
2531

2632
def __init__(self,
2733
*,
28-
status : typing___Optional[int] = None,
34+
status : typing___Optional[builtin___int] = None,
2935
message : typing___Optional[typing___Text] = None,
3036
) -> None: ...
3137
@classmethod
32-
def FromString(cls, s: bytes) -> HeaderProto: ...
38+
def FromString(cls, s: builtin___bytes) -> HeaderProto: ...
3339
def MergeFrom(self, other_msg: google___protobuf___message___Message) -> None: ...
3440
def CopyFrom(self, other_msg: google___protobuf___message___Message) -> None: ...
3541
if sys.version_info >= (3,):

0 commit comments

Comments
 (0)