luckyrobots 0.1.70__py3-none-any.whl → 0.1.71__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- luckyrobots/__init__.py +8 -1
- luckyrobots/client.py +98 -4
- luckyrobots/grpc/generated/agent_pb2.py +36 -32
- luckyrobots/grpc/generated/agent_pb2_grpc.py +110 -128
- luckyrobots/grpc/generated/camera_pb2.py +20 -18
- luckyrobots/grpc/generated/camera_pb2_grpc.py +62 -73
- luckyrobots/grpc/generated/common_pb2.py +20 -16
- luckyrobots/grpc/generated/common_pb2_grpc.py +7 -10
- luckyrobots/grpc/generated/hazel_rpc_pb2.py +43 -0
- luckyrobots/grpc/generated/hazel_rpc_pb2_grpc.py +24 -0
- luckyrobots/grpc/generated/media_pb2.py +16 -12
- luckyrobots/grpc/generated/media_pb2_grpc.py +7 -10
- luckyrobots/grpc/generated/mujoco_pb2.py +28 -24
- luckyrobots/grpc/generated/mujoco_pb2_grpc.py +110 -128
- luckyrobots/grpc/generated/scene_pb2.py +30 -28
- luckyrobots/grpc/generated/scene_pb2_grpc.py +108 -127
- luckyrobots/grpc/generated/telemetry_pb2.py +24 -20
- luckyrobots/grpc/generated/telemetry_pb2_grpc.py +62 -73
- luckyrobots/grpc/generated/viewport_pb2.py +24 -22
- luckyrobots/grpc/generated/viewport_pb2_grpc.py +62 -73
- luckyrobots/grpc/proto/agent.proto +33 -1
- luckyrobots/grpc/proto/camera.proto +1 -1
- luckyrobots/grpc/proto/common.proto +1 -1
- luckyrobots/grpc/proto/hazel_rpc.proto +1 -1
- luckyrobots/grpc/proto/media.proto +1 -1
- luckyrobots/grpc/proto/mujoco.proto +1 -1
- luckyrobots/grpc/proto/scene.proto +1 -1
- luckyrobots/grpc/proto/telemetry.proto +1 -1
- luckyrobots/grpc/proto/viewport.proto +1 -1
- luckyrobots/luckyrobots.py +11 -2
- luckyrobots/models/__init__.py +2 -0
- luckyrobots/models/randomization.py +77 -0
- {luckyrobots-0.1.70.dist-info → luckyrobots-0.1.71.dist-info}/METADATA +1 -1
- luckyrobots-0.1.71.dist-info/RECORD +47 -0
- luckyrobots-0.1.70.dist-info/RECORD +0 -44
- {luckyrobots-0.1.70.dist-info → luckyrobots-0.1.71.dist-info}/WHEEL +0 -0
- {luckyrobots-0.1.70.dist-info → luckyrobots-0.1.71.dist-info}/licenses/LICENSE +0 -0
|
@@ -5,31 +5,29 @@ import warnings
|
|
|
5
5
|
|
|
6
6
|
from . import scene_pb2 as scene__pb2
|
|
7
7
|
|
|
8
|
-
GRPC_GENERATED_VERSION =
|
|
8
|
+
GRPC_GENERATED_VERSION = '1.76.0'
|
|
9
9
|
GRPC_VERSION = grpc.__version__
|
|
10
10
|
_version_not_supported = False
|
|
11
11
|
|
|
12
12
|
try:
|
|
13
13
|
from grpc._utilities import first_version_is_lower
|
|
14
|
-
|
|
15
|
-
_version_not_supported = first_version_is_lower(
|
|
16
|
-
GRPC_VERSION, GRPC_GENERATED_VERSION
|
|
17
|
-
)
|
|
14
|
+
_version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
|
|
18
15
|
except ImportError:
|
|
19
16
|
_version_not_supported = True
|
|
20
17
|
|
|
21
18
|
if _version_not_supported:
|
|
22
19
|
raise RuntimeError(
|
|
23
|
-
f
|
|
24
|
-
+
|
|
25
|
-
+ f
|
|
26
|
-
+ f
|
|
27
|
-
+ f
|
|
20
|
+
f'The grpc package installed is at version {GRPC_VERSION},'
|
|
21
|
+
+ ' but the generated code in scene_pb2_grpc.py depends on'
|
|
22
|
+
+ f' grpcio>={GRPC_GENERATED_VERSION}.'
|
|
23
|
+
+ f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
|
|
24
|
+
+ f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
|
|
28
25
|
)
|
|
29
26
|
|
|
30
27
|
|
|
31
28
|
class SceneServiceStub(object):
|
|
32
|
-
"""Scene inspection + basic editing.
|
|
29
|
+
"""Scene inspection + basic editing.
|
|
30
|
+
"""
|
|
33
31
|
|
|
34
32
|
def __init__(self, channel):
|
|
35
33
|
"""Constructor.
|
|
@@ -38,112 +36,105 @@ class SceneServiceStub(object):
|
|
|
38
36
|
channel: A grpc.Channel.
|
|
39
37
|
"""
|
|
40
38
|
self.GetSceneInfo = channel.unary_unary(
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
)
|
|
39
|
+
'/hazel.rpc.SceneService/GetSceneInfo',
|
|
40
|
+
request_serializer=scene__pb2.GetSceneInfoRequest.SerializeToString,
|
|
41
|
+
response_deserializer=scene__pb2.GetSceneInfoResponse.FromString,
|
|
42
|
+
_registered_method=True)
|
|
46
43
|
self.ListEntities = channel.unary_unary(
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
)
|
|
44
|
+
'/hazel.rpc.SceneService/ListEntities',
|
|
45
|
+
request_serializer=scene__pb2.ListEntitiesRequest.SerializeToString,
|
|
46
|
+
response_deserializer=scene__pb2.ListEntitiesResponse.FromString,
|
|
47
|
+
_registered_method=True)
|
|
52
48
|
self.GetEntity = channel.unary_unary(
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
)
|
|
49
|
+
'/hazel.rpc.SceneService/GetEntity',
|
|
50
|
+
request_serializer=scene__pb2.GetEntityRequest.SerializeToString,
|
|
51
|
+
response_deserializer=scene__pb2.GetEntityResponse.FromString,
|
|
52
|
+
_registered_method=True)
|
|
58
53
|
self.SetEntityTransform = channel.unary_unary(
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
)
|
|
54
|
+
'/hazel.rpc.SceneService/SetEntityTransform',
|
|
55
|
+
request_serializer=scene__pb2.SetEntityTransformRequest.SerializeToString,
|
|
56
|
+
response_deserializer=scene__pb2.SetEntityTransformResponse.FromString,
|
|
57
|
+
_registered_method=True)
|
|
64
58
|
|
|
65
59
|
|
|
66
60
|
class SceneServiceServicer(object):
|
|
67
|
-
"""Scene inspection + basic editing.
|
|
61
|
+
"""Scene inspection + basic editing.
|
|
62
|
+
"""
|
|
68
63
|
|
|
69
64
|
def GetSceneInfo(self, request, context):
|
|
70
65
|
"""Missing associated documentation comment in .proto file."""
|
|
71
66
|
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
72
|
-
context.set_details(
|
|
73
|
-
raise NotImplementedError(
|
|
67
|
+
context.set_details('Method not implemented!')
|
|
68
|
+
raise NotImplementedError('Method not implemented!')
|
|
74
69
|
|
|
75
70
|
def ListEntities(self, request, context):
|
|
76
71
|
"""Missing associated documentation comment in .proto file."""
|
|
77
72
|
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
78
|
-
context.set_details(
|
|
79
|
-
raise NotImplementedError(
|
|
73
|
+
context.set_details('Method not implemented!')
|
|
74
|
+
raise NotImplementedError('Method not implemented!')
|
|
80
75
|
|
|
81
76
|
def GetEntity(self, request, context):
|
|
82
77
|
"""Missing associated documentation comment in .proto file."""
|
|
83
78
|
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
84
|
-
context.set_details(
|
|
85
|
-
raise NotImplementedError(
|
|
79
|
+
context.set_details('Method not implemented!')
|
|
80
|
+
raise NotImplementedError('Method not implemented!')
|
|
86
81
|
|
|
87
82
|
def SetEntityTransform(self, request, context):
|
|
88
83
|
"""Missing associated documentation comment in .proto file."""
|
|
89
84
|
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
90
|
-
context.set_details(
|
|
91
|
-
raise NotImplementedError(
|
|
85
|
+
context.set_details('Method not implemented!')
|
|
86
|
+
raise NotImplementedError('Method not implemented!')
|
|
92
87
|
|
|
93
88
|
|
|
94
89
|
def add_SceneServiceServicer_to_server(servicer, server):
|
|
95
90
|
rpc_method_handlers = {
|
|
96
|
-
|
|
97
|
-
|
|
98
|
-
|
|
99
|
-
|
|
100
|
-
|
|
101
|
-
|
|
102
|
-
|
|
103
|
-
|
|
104
|
-
|
|
105
|
-
|
|
106
|
-
|
|
107
|
-
|
|
108
|
-
|
|
109
|
-
|
|
110
|
-
|
|
111
|
-
|
|
112
|
-
|
|
113
|
-
|
|
114
|
-
|
|
115
|
-
|
|
91
|
+
'GetSceneInfo': grpc.unary_unary_rpc_method_handler(
|
|
92
|
+
servicer.GetSceneInfo,
|
|
93
|
+
request_deserializer=scene__pb2.GetSceneInfoRequest.FromString,
|
|
94
|
+
response_serializer=scene__pb2.GetSceneInfoResponse.SerializeToString,
|
|
95
|
+
),
|
|
96
|
+
'ListEntities': grpc.unary_unary_rpc_method_handler(
|
|
97
|
+
servicer.ListEntities,
|
|
98
|
+
request_deserializer=scene__pb2.ListEntitiesRequest.FromString,
|
|
99
|
+
response_serializer=scene__pb2.ListEntitiesResponse.SerializeToString,
|
|
100
|
+
),
|
|
101
|
+
'GetEntity': grpc.unary_unary_rpc_method_handler(
|
|
102
|
+
servicer.GetEntity,
|
|
103
|
+
request_deserializer=scene__pb2.GetEntityRequest.FromString,
|
|
104
|
+
response_serializer=scene__pb2.GetEntityResponse.SerializeToString,
|
|
105
|
+
),
|
|
106
|
+
'SetEntityTransform': grpc.unary_unary_rpc_method_handler(
|
|
107
|
+
servicer.SetEntityTransform,
|
|
108
|
+
request_deserializer=scene__pb2.SetEntityTransformRequest.FromString,
|
|
109
|
+
response_serializer=scene__pb2.SetEntityTransformResponse.SerializeToString,
|
|
110
|
+
),
|
|
116
111
|
}
|
|
117
112
|
generic_handler = grpc.method_handlers_generic_handler(
|
|
118
|
-
|
|
119
|
-
)
|
|
113
|
+
'hazel.rpc.SceneService', rpc_method_handlers)
|
|
120
114
|
server.add_generic_rpc_handlers((generic_handler,))
|
|
121
|
-
server.add_registered_method_handlers(
|
|
122
|
-
"hazel.rpc.v1.SceneService", rpc_method_handlers
|
|
123
|
-
)
|
|
115
|
+
server.add_registered_method_handlers('hazel.rpc.SceneService', rpc_method_handlers)
|
|
124
116
|
|
|
125
117
|
|
|
126
|
-
# This class is part of an EXPERIMENTAL API.
|
|
118
|
+
# This class is part of an EXPERIMENTAL API.
|
|
127
119
|
class SceneService(object):
|
|
128
|
-
"""Scene inspection + basic editing.
|
|
120
|
+
"""Scene inspection + basic editing.
|
|
121
|
+
"""
|
|
129
122
|
|
|
130
123
|
@staticmethod
|
|
131
|
-
def GetSceneInfo(
|
|
132
|
-
|
|
133
|
-
|
|
134
|
-
|
|
135
|
-
|
|
136
|
-
|
|
137
|
-
|
|
138
|
-
|
|
139
|
-
|
|
140
|
-
|
|
141
|
-
metadata=None,
|
|
142
|
-
):
|
|
124
|
+
def GetSceneInfo(request,
|
|
125
|
+
target,
|
|
126
|
+
options=(),
|
|
127
|
+
channel_credentials=None,
|
|
128
|
+
call_credentials=None,
|
|
129
|
+
insecure=False,
|
|
130
|
+
compression=None,
|
|
131
|
+
wait_for_ready=None,
|
|
132
|
+
timeout=None,
|
|
133
|
+
metadata=None):
|
|
143
134
|
return grpc.experimental.unary_unary(
|
|
144
135
|
request,
|
|
145
136
|
target,
|
|
146
|
-
|
|
137
|
+
'/hazel.rpc.SceneService/GetSceneInfo',
|
|
147
138
|
scene__pb2.GetSceneInfoRequest.SerializeToString,
|
|
148
139
|
scene__pb2.GetSceneInfoResponse.FromString,
|
|
149
140
|
options,
|
|
@@ -154,26 +145,23 @@ class SceneService(object):
|
|
|
154
145
|
wait_for_ready,
|
|
155
146
|
timeout,
|
|
156
147
|
metadata,
|
|
157
|
-
_registered_method=True
|
|
158
|
-
)
|
|
148
|
+
_registered_method=True)
|
|
159
149
|
|
|
160
150
|
@staticmethod
|
|
161
|
-
def ListEntities(
|
|
162
|
-
|
|
163
|
-
|
|
164
|
-
|
|
165
|
-
|
|
166
|
-
|
|
167
|
-
|
|
168
|
-
|
|
169
|
-
|
|
170
|
-
|
|
171
|
-
metadata=None,
|
|
172
|
-
):
|
|
151
|
+
def ListEntities(request,
|
|
152
|
+
target,
|
|
153
|
+
options=(),
|
|
154
|
+
channel_credentials=None,
|
|
155
|
+
call_credentials=None,
|
|
156
|
+
insecure=False,
|
|
157
|
+
compression=None,
|
|
158
|
+
wait_for_ready=None,
|
|
159
|
+
timeout=None,
|
|
160
|
+
metadata=None):
|
|
173
161
|
return grpc.experimental.unary_unary(
|
|
174
162
|
request,
|
|
175
163
|
target,
|
|
176
|
-
|
|
164
|
+
'/hazel.rpc.SceneService/ListEntities',
|
|
177
165
|
scene__pb2.ListEntitiesRequest.SerializeToString,
|
|
178
166
|
scene__pb2.ListEntitiesResponse.FromString,
|
|
179
167
|
options,
|
|
@@ -184,26 +172,23 @@ class SceneService(object):
|
|
|
184
172
|
wait_for_ready,
|
|
185
173
|
timeout,
|
|
186
174
|
metadata,
|
|
187
|
-
_registered_method=True
|
|
188
|
-
)
|
|
175
|
+
_registered_method=True)
|
|
189
176
|
|
|
190
177
|
@staticmethod
|
|
191
|
-
def GetEntity(
|
|
192
|
-
|
|
193
|
-
|
|
194
|
-
|
|
195
|
-
|
|
196
|
-
|
|
197
|
-
|
|
198
|
-
|
|
199
|
-
|
|
200
|
-
|
|
201
|
-
metadata=None,
|
|
202
|
-
):
|
|
178
|
+
def GetEntity(request,
|
|
179
|
+
target,
|
|
180
|
+
options=(),
|
|
181
|
+
channel_credentials=None,
|
|
182
|
+
call_credentials=None,
|
|
183
|
+
insecure=False,
|
|
184
|
+
compression=None,
|
|
185
|
+
wait_for_ready=None,
|
|
186
|
+
timeout=None,
|
|
187
|
+
metadata=None):
|
|
203
188
|
return grpc.experimental.unary_unary(
|
|
204
189
|
request,
|
|
205
190
|
target,
|
|
206
|
-
|
|
191
|
+
'/hazel.rpc.SceneService/GetEntity',
|
|
207
192
|
scene__pb2.GetEntityRequest.SerializeToString,
|
|
208
193
|
scene__pb2.GetEntityResponse.FromString,
|
|
209
194
|
options,
|
|
@@ -214,26 +199,23 @@ class SceneService(object):
|
|
|
214
199
|
wait_for_ready,
|
|
215
200
|
timeout,
|
|
216
201
|
metadata,
|
|
217
|
-
_registered_method=True
|
|
218
|
-
)
|
|
202
|
+
_registered_method=True)
|
|
219
203
|
|
|
220
204
|
@staticmethod
|
|
221
|
-
def SetEntityTransform(
|
|
222
|
-
|
|
223
|
-
|
|
224
|
-
|
|
225
|
-
|
|
226
|
-
|
|
227
|
-
|
|
228
|
-
|
|
229
|
-
|
|
230
|
-
|
|
231
|
-
metadata=None,
|
|
232
|
-
):
|
|
205
|
+
def SetEntityTransform(request,
|
|
206
|
+
target,
|
|
207
|
+
options=(),
|
|
208
|
+
channel_credentials=None,
|
|
209
|
+
call_credentials=None,
|
|
210
|
+
insecure=False,
|
|
211
|
+
compression=None,
|
|
212
|
+
wait_for_ready=None,
|
|
213
|
+
timeout=None,
|
|
214
|
+
metadata=None):
|
|
233
215
|
return grpc.experimental.unary_unary(
|
|
234
216
|
request,
|
|
235
217
|
target,
|
|
236
|
-
|
|
218
|
+
'/hazel.rpc.SceneService/SetEntityTransform',
|
|
237
219
|
scene__pb2.SetEntityTransformRequest.SerializeToString,
|
|
238
220
|
scene__pb2.SetEntityTransformResponse.FromString,
|
|
239
221
|
options,
|
|
@@ -244,5 +226,4 @@ class SceneService(object):
|
|
|
244
226
|
wait_for_ready,
|
|
245
227
|
timeout,
|
|
246
228
|
metadata,
|
|
247
|
-
_registered_method=True
|
|
248
|
-
)
|
|
229
|
+
_registered_method=True)
|
|
@@ -9,35 +9,39 @@ from google.protobuf import descriptor_pool as _descriptor_pool
|
|
|
9
9
|
from google.protobuf import runtime_version as _runtime_version
|
|
10
10
|
from google.protobuf import symbol_database as _symbol_database
|
|
11
11
|
from google.protobuf.internal import builder as _builder
|
|
12
|
-
|
|
13
12
|
_runtime_version.ValidateProtobufRuntimeVersion(
|
|
14
|
-
_runtime_version.Domain.PUBLIC,
|
|
13
|
+
_runtime_version.Domain.PUBLIC,
|
|
14
|
+
6,
|
|
15
|
+
31,
|
|
16
|
+
1,
|
|
17
|
+
'',
|
|
18
|
+
'telemetry.proto'
|
|
15
19
|
)
|
|
16
20
|
# @@protoc_insertion_point(imports)
|
|
17
21
|
|
|
18
22
|
_sym_db = _symbol_database.Default()
|
|
19
23
|
|
|
20
24
|
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
)
|
|
25
|
+
|
|
26
|
+
|
|
27
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0ftelemetry.proto\x12\thazel.rpc\"Z\n\x0fTelemetrySchema\x12\x19\n\x11observation_names\x18\x01 \x03(\t\x12\x14\n\x0c\x61\x63tion_names\x18\x02 \x03(\t\x12\n\n\x02nq\x18\x03 \x01(\r\x12\n\n\x02nu\x18\x04 \x01(\r\"\x1b\n\x19GetTelemetrySchemaRequest\"H\n\x1aGetTelemetrySchemaResponse\x12*\n\x06schema\x18\x01 \x01(\x0b\x32\x1a.hazel.rpc.TelemetrySchema\",\n\x16StreamTelemetryRequest\x12\x12\n\ntarget_fps\x18\x01 \x01(\r\"\x7f\n\x0eTelemetryFrame\x12\x14\n\x0ctimestamp_ms\x18\x01 \x01(\x04\x12\x14\n\x0c\x66rame_number\x18\x02 \x01(\r\x12\x12\n\ntask_index\x18\x03 \x01(\r\x12\x18\n\x10observation_qpos\x18\x04 \x03(\x02\x12\x13\n\x0b\x61\x63tion_ctrl\x18\x05 \x03(\x02\x32\xc8\x01\n\x10TelemetryService\x12\x61\n\x12GetTelemetrySchema\x12$.hazel.rpc.GetTelemetrySchemaRequest\x1a%.hazel.rpc.GetTelemetrySchemaResponse\x12Q\n\x0fStreamTelemetry\x12!.hazel.rpc.StreamTelemetryRequest\x1a\x19.hazel.rpc.TelemetryFrame0\x01\x42\x03\xf8\x01\x01\x62\x06proto3')
|
|
24
28
|
|
|
25
29
|
_globals = globals()
|
|
26
30
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
|
27
|
-
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR,
|
|
31
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'telemetry_pb2', _globals)
|
|
28
32
|
if not _descriptor._USE_C_DESCRIPTORS:
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
33
|
+
_globals['DESCRIPTOR']._loaded_options = None
|
|
34
|
+
_globals['DESCRIPTOR']._serialized_options = b'\370\001\001'
|
|
35
|
+
_globals['_TELEMETRYSCHEMA']._serialized_start=30
|
|
36
|
+
_globals['_TELEMETRYSCHEMA']._serialized_end=120
|
|
37
|
+
_globals['_GETTELEMETRYSCHEMAREQUEST']._serialized_start=122
|
|
38
|
+
_globals['_GETTELEMETRYSCHEMAREQUEST']._serialized_end=149
|
|
39
|
+
_globals['_GETTELEMETRYSCHEMARESPONSE']._serialized_start=151
|
|
40
|
+
_globals['_GETTELEMETRYSCHEMARESPONSE']._serialized_end=223
|
|
41
|
+
_globals['_STREAMTELEMETRYREQUEST']._serialized_start=225
|
|
42
|
+
_globals['_STREAMTELEMETRYREQUEST']._serialized_end=269
|
|
43
|
+
_globals['_TELEMETRYFRAME']._serialized_start=271
|
|
44
|
+
_globals['_TELEMETRYFRAME']._serialized_end=398
|
|
45
|
+
_globals['_TELEMETRYSERVICE']._serialized_start=401
|
|
46
|
+
_globals['_TELEMETRYSERVICE']._serialized_end=601
|
|
43
47
|
# @@protoc_insertion_point(module_scope)
|
|
@@ -5,31 +5,29 @@ import warnings
|
|
|
5
5
|
|
|
6
6
|
from . import telemetry_pb2 as telemetry__pb2
|
|
7
7
|
|
|
8
|
-
GRPC_GENERATED_VERSION =
|
|
8
|
+
GRPC_GENERATED_VERSION = '1.76.0'
|
|
9
9
|
GRPC_VERSION = grpc.__version__
|
|
10
10
|
_version_not_supported = False
|
|
11
11
|
|
|
12
12
|
try:
|
|
13
13
|
from grpc._utilities import first_version_is_lower
|
|
14
|
-
|
|
15
|
-
_version_not_supported = first_version_is_lower(
|
|
16
|
-
GRPC_VERSION, GRPC_GENERATED_VERSION
|
|
17
|
-
)
|
|
14
|
+
_version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
|
|
18
15
|
except ImportError:
|
|
19
16
|
_version_not_supported = True
|
|
20
17
|
|
|
21
18
|
if _version_not_supported:
|
|
22
19
|
raise RuntimeError(
|
|
23
|
-
f
|
|
24
|
-
+
|
|
25
|
-
+ f
|
|
26
|
-
+ f
|
|
27
|
-
+ f
|
|
20
|
+
f'The grpc package installed is at version {GRPC_VERSION},'
|
|
21
|
+
+ ' but the generated code in telemetry_pb2_grpc.py depends on'
|
|
22
|
+
+ f' grpcio>={GRPC_GENERATED_VERSION}.'
|
|
23
|
+
+ f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
|
|
24
|
+
+ f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
|
|
28
25
|
)
|
|
29
26
|
|
|
30
27
|
|
|
31
28
|
class TelemetryServiceStub(object):
|
|
32
|
-
"""Stream-only service intended for external tools (plotting, training logs, debugging).
|
|
29
|
+
"""Stream-only service intended for external tools (plotting, training logs, debugging).
|
|
30
|
+
"""
|
|
33
31
|
|
|
34
32
|
def __init__(self, channel):
|
|
35
33
|
"""Constructor.
|
|
@@ -38,78 +36,73 @@ class TelemetryServiceStub(object):
|
|
|
38
36
|
channel: A grpc.Channel.
|
|
39
37
|
"""
|
|
40
38
|
self.GetTelemetrySchema = channel.unary_unary(
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
)
|
|
39
|
+
'/hazel.rpc.TelemetryService/GetTelemetrySchema',
|
|
40
|
+
request_serializer=telemetry__pb2.GetTelemetrySchemaRequest.SerializeToString,
|
|
41
|
+
response_deserializer=telemetry__pb2.GetTelemetrySchemaResponse.FromString,
|
|
42
|
+
_registered_method=True)
|
|
46
43
|
self.StreamTelemetry = channel.unary_stream(
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
)
|
|
44
|
+
'/hazel.rpc.TelemetryService/StreamTelemetry',
|
|
45
|
+
request_serializer=telemetry__pb2.StreamTelemetryRequest.SerializeToString,
|
|
46
|
+
response_deserializer=telemetry__pb2.TelemetryFrame.FromString,
|
|
47
|
+
_registered_method=True)
|
|
52
48
|
|
|
53
49
|
|
|
54
50
|
class TelemetryServiceServicer(object):
|
|
55
|
-
"""Stream-only service intended for external tools (plotting, training logs, debugging).
|
|
51
|
+
"""Stream-only service intended for external tools (plotting, training logs, debugging).
|
|
52
|
+
"""
|
|
56
53
|
|
|
57
54
|
def GetTelemetrySchema(self, request, context):
|
|
58
55
|
"""Missing associated documentation comment in .proto file."""
|
|
59
56
|
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
60
|
-
context.set_details(
|
|
61
|
-
raise NotImplementedError(
|
|
57
|
+
context.set_details('Method not implemented!')
|
|
58
|
+
raise NotImplementedError('Method not implemented!')
|
|
62
59
|
|
|
63
60
|
def StreamTelemetry(self, request, context):
|
|
64
61
|
"""Missing associated documentation comment in .proto file."""
|
|
65
62
|
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
66
|
-
context.set_details(
|
|
67
|
-
raise NotImplementedError(
|
|
63
|
+
context.set_details('Method not implemented!')
|
|
64
|
+
raise NotImplementedError('Method not implemented!')
|
|
68
65
|
|
|
69
66
|
|
|
70
67
|
def add_TelemetryServiceServicer_to_server(servicer, server):
|
|
71
68
|
rpc_method_handlers = {
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
|
|
80
|
-
|
|
81
|
-
|
|
69
|
+
'GetTelemetrySchema': grpc.unary_unary_rpc_method_handler(
|
|
70
|
+
servicer.GetTelemetrySchema,
|
|
71
|
+
request_deserializer=telemetry__pb2.GetTelemetrySchemaRequest.FromString,
|
|
72
|
+
response_serializer=telemetry__pb2.GetTelemetrySchemaResponse.SerializeToString,
|
|
73
|
+
),
|
|
74
|
+
'StreamTelemetry': grpc.unary_stream_rpc_method_handler(
|
|
75
|
+
servicer.StreamTelemetry,
|
|
76
|
+
request_deserializer=telemetry__pb2.StreamTelemetryRequest.FromString,
|
|
77
|
+
response_serializer=telemetry__pb2.TelemetryFrame.SerializeToString,
|
|
78
|
+
),
|
|
82
79
|
}
|
|
83
80
|
generic_handler = grpc.method_handlers_generic_handler(
|
|
84
|
-
|
|
85
|
-
)
|
|
81
|
+
'hazel.rpc.TelemetryService', rpc_method_handlers)
|
|
86
82
|
server.add_generic_rpc_handlers((generic_handler,))
|
|
87
|
-
server.add_registered_method_handlers(
|
|
88
|
-
"hazel.rpc.v1.TelemetryService", rpc_method_handlers
|
|
89
|
-
)
|
|
83
|
+
server.add_registered_method_handlers('hazel.rpc.TelemetryService', rpc_method_handlers)
|
|
90
84
|
|
|
91
85
|
|
|
92
|
-
# This class is part of an EXPERIMENTAL API.
|
|
86
|
+
# This class is part of an EXPERIMENTAL API.
|
|
93
87
|
class TelemetryService(object):
|
|
94
|
-
"""Stream-only service intended for external tools (plotting, training logs, debugging).
|
|
88
|
+
"""Stream-only service intended for external tools (plotting, training logs, debugging).
|
|
89
|
+
"""
|
|
95
90
|
|
|
96
91
|
@staticmethod
|
|
97
|
-
def GetTelemetrySchema(
|
|
98
|
-
|
|
99
|
-
|
|
100
|
-
|
|
101
|
-
|
|
102
|
-
|
|
103
|
-
|
|
104
|
-
|
|
105
|
-
|
|
106
|
-
|
|
107
|
-
metadata=None,
|
|
108
|
-
):
|
|
92
|
+
def GetTelemetrySchema(request,
|
|
93
|
+
target,
|
|
94
|
+
options=(),
|
|
95
|
+
channel_credentials=None,
|
|
96
|
+
call_credentials=None,
|
|
97
|
+
insecure=False,
|
|
98
|
+
compression=None,
|
|
99
|
+
wait_for_ready=None,
|
|
100
|
+
timeout=None,
|
|
101
|
+
metadata=None):
|
|
109
102
|
return grpc.experimental.unary_unary(
|
|
110
103
|
request,
|
|
111
104
|
target,
|
|
112
|
-
|
|
105
|
+
'/hazel.rpc.TelemetryService/GetTelemetrySchema',
|
|
113
106
|
telemetry__pb2.GetTelemetrySchemaRequest.SerializeToString,
|
|
114
107
|
telemetry__pb2.GetTelemetrySchemaResponse.FromString,
|
|
115
108
|
options,
|
|
@@ -120,26 +113,23 @@ class TelemetryService(object):
|
|
|
120
113
|
wait_for_ready,
|
|
121
114
|
timeout,
|
|
122
115
|
metadata,
|
|
123
|
-
_registered_method=True
|
|
124
|
-
)
|
|
116
|
+
_registered_method=True)
|
|
125
117
|
|
|
126
118
|
@staticmethod
|
|
127
|
-
def StreamTelemetry(
|
|
128
|
-
|
|
129
|
-
|
|
130
|
-
|
|
131
|
-
|
|
132
|
-
|
|
133
|
-
|
|
134
|
-
|
|
135
|
-
|
|
136
|
-
|
|
137
|
-
metadata=None,
|
|
138
|
-
):
|
|
119
|
+
def StreamTelemetry(request,
|
|
120
|
+
target,
|
|
121
|
+
options=(),
|
|
122
|
+
channel_credentials=None,
|
|
123
|
+
call_credentials=None,
|
|
124
|
+
insecure=False,
|
|
125
|
+
compression=None,
|
|
126
|
+
wait_for_ready=None,
|
|
127
|
+
timeout=None,
|
|
128
|
+
metadata=None):
|
|
139
129
|
return grpc.experimental.unary_stream(
|
|
140
130
|
request,
|
|
141
131
|
target,
|
|
142
|
-
|
|
132
|
+
'/hazel.rpc.TelemetryService/StreamTelemetry',
|
|
143
133
|
telemetry__pb2.StreamTelemetryRequest.SerializeToString,
|
|
144
134
|
telemetry__pb2.TelemetryFrame.FromString,
|
|
145
135
|
options,
|
|
@@ -150,5 +140,4 @@ class TelemetryService(object):
|
|
|
150
140
|
wait_for_ready,
|
|
151
141
|
timeout,
|
|
152
142
|
metadata,
|
|
153
|
-
_registered_method=True
|
|
154
|
-
)
|
|
143
|
+
_registered_method=True)
|