kinfer 0.3.3__cp312-cp312-macosx_11_0_arm64.whl → 0.4.1__cp312-cp312-macosx_11_0_arm64.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.
Files changed (45) hide show
  1. kinfer/__init__.py +0 -5
  2. kinfer/common/__init__.py +0 -0
  3. kinfer/common/types.py +12 -0
  4. kinfer/export/common.py +41 -0
  5. kinfer/export/jax.py +53 -0
  6. kinfer/export/pytorch.py +45 -110
  7. kinfer/export/serialize.py +93 -0
  8. kinfer/requirements.txt +3 -4
  9. kinfer/rust/Cargo.toml +20 -8
  10. kinfer/rust/src/lib.rs +2 -11
  11. kinfer/rust/src/model.rs +286 -121
  12. kinfer/rust/src/runtime.rs +104 -0
  13. kinfer/rust_bindings/Cargo.toml +8 -1
  14. kinfer/rust_bindings/rust_bindings.pyi +36 -0
  15. kinfer/rust_bindings/src/lib.rs +326 -1
  16. kinfer/rust_bindings.cpython-312-darwin.so +0 -0
  17. kinfer/rust_bindings.pyi +30 -1
  18. kinfer-0.4.1.dist-info/METADATA +55 -0
  19. kinfer-0.4.1.dist-info/RECORD +26 -0
  20. {kinfer-0.3.3.dist-info → kinfer-0.4.1.dist-info}/WHEEL +2 -1
  21. kinfer/inference/__init__.py +0 -2
  22. kinfer/inference/base.py +0 -64
  23. kinfer/inference/python.py +0 -66
  24. kinfer/proto/__init__.py +0 -40
  25. kinfer/proto/kinfer_pb2.py +0 -103
  26. kinfer/proto/kinfer_pb2.pyi +0 -1097
  27. kinfer/requirements-dev.txt +0 -8
  28. kinfer/rust/build.rs +0 -16
  29. kinfer/rust/src/kinfer_proto.rs +0 -14
  30. kinfer/rust/src/main.rs +0 -6
  31. kinfer/rust/src/onnx_serializer.rs +0 -804
  32. kinfer/rust/src/serializer.rs +0 -221
  33. kinfer/rust/src/tests/onnx_serializer_tests.rs +0 -212
  34. kinfer/serialize/__init__.py +0 -60
  35. kinfer/serialize/base.py +0 -536
  36. kinfer/serialize/json.py +0 -399
  37. kinfer/serialize/numpy.py +0 -426
  38. kinfer/serialize/pytorch.py +0 -402
  39. kinfer/serialize/schema.py +0 -125
  40. kinfer/serialize/types.py +0 -17
  41. kinfer/serialize/utils.py +0 -177
  42. kinfer-0.3.3.dist-info/METADATA +0 -57
  43. kinfer-0.3.3.dist-info/RECORD +0 -40
  44. {kinfer-0.3.3.dist-info → kinfer-0.4.1.dist-info/licenses}/LICENSE +0 -0
  45. {kinfer-0.3.3.dist-info → kinfer-0.4.1.dist-info}/top_level.txt +0 -0
@@ -1,1097 +0,0 @@
1
- """
2
- @generated by mypy-protobuf. Do not edit manually!
3
- isort:skip_file
4
- """
5
-
6
- import builtins
7
- import collections.abc
8
- import google.protobuf.descriptor
9
- import google.protobuf.internal.containers
10
- import google.protobuf.internal.enum_type_wrapper
11
- import google.protobuf.message
12
- import sys
13
- import typing
14
-
15
- if sys.version_info >= (3, 10):
16
- import typing as typing_extensions
17
- else:
18
- import typing_extensions
19
-
20
- DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
21
-
22
- class _DType:
23
- ValueType = typing.NewType("ValueType", builtins.int)
24
- V: typing_extensions.TypeAlias = ValueType
25
-
26
- class _DTypeEnumTypeWrapper(
27
- google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_DType.ValueType], builtins.type
28
- ):
29
- DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
30
- FP8: _DType.ValueType # 0
31
- FP16: _DType.ValueType # 1
32
- FP32: _DType.ValueType # 2
33
- FP64: _DType.ValueType # 3
34
- INT8: _DType.ValueType # 4
35
- INT16: _DType.ValueType # 5
36
- INT32: _DType.ValueType # 6
37
- INT64: _DType.ValueType # 7
38
- UINT8: _DType.ValueType # 8
39
- UINT16: _DType.ValueType # 9
40
- UINT32: _DType.ValueType # 10
41
- UINT64: _DType.ValueType # 11
42
-
43
- class DType(_DType, metaclass=_DTypeEnumTypeWrapper):
44
- """----- *
45
- Types *
46
- -----
47
- """
48
-
49
- FP8: DType.ValueType # 0
50
- FP16: DType.ValueType # 1
51
- FP32: DType.ValueType # 2
52
- FP64: DType.ValueType # 3
53
- INT8: DType.ValueType # 4
54
- INT16: DType.ValueType # 5
55
- INT32: DType.ValueType # 6
56
- INT64: DType.ValueType # 7
57
- UINT8: DType.ValueType # 8
58
- UINT16: DType.ValueType # 9
59
- UINT32: DType.ValueType # 10
60
- UINT64: DType.ValueType # 11
61
- global___DType = DType
62
-
63
- class _JointPositionUnit:
64
- ValueType = typing.NewType("ValueType", builtins.int)
65
- V: typing_extensions.TypeAlias = ValueType
66
-
67
- class _JointPositionUnitEnumTypeWrapper(
68
- google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_JointPositionUnit.ValueType], builtins.type
69
- ):
70
- DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
71
- DEGREES: _JointPositionUnit.ValueType # 0
72
- RADIANS: _JointPositionUnit.ValueType # 1
73
-
74
- class JointPositionUnit(_JointPositionUnit, metaclass=_JointPositionUnitEnumTypeWrapper):
75
- """--------------- *
76
- Joint Positions *
77
- ---------------
78
- """
79
-
80
- DEGREES: JointPositionUnit.ValueType # 0
81
- RADIANS: JointPositionUnit.ValueType # 1
82
- global___JointPositionUnit = JointPositionUnit
83
-
84
- class _JointVelocityUnit:
85
- ValueType = typing.NewType("ValueType", builtins.int)
86
- V: typing_extensions.TypeAlias = ValueType
87
-
88
- class _JointVelocityUnitEnumTypeWrapper(
89
- google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_JointVelocityUnit.ValueType], builtins.type
90
- ):
91
- DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
92
- DEGREES_PER_SECOND: _JointVelocityUnit.ValueType # 0
93
- RADIANS_PER_SECOND: _JointVelocityUnit.ValueType # 1
94
-
95
- class JointVelocityUnit(_JointVelocityUnit, metaclass=_JointVelocityUnitEnumTypeWrapper):
96
- """---------------- *
97
- Joint Velocities *
98
- ----------------
99
- """
100
-
101
- DEGREES_PER_SECOND: JointVelocityUnit.ValueType # 0
102
- RADIANS_PER_SECOND: JointVelocityUnit.ValueType # 1
103
- global___JointVelocityUnit = JointVelocityUnit
104
-
105
- class _JointTorqueUnit:
106
- ValueType = typing.NewType("ValueType", builtins.int)
107
- V: typing_extensions.TypeAlias = ValueType
108
-
109
- class _JointTorqueUnitEnumTypeWrapper(
110
- google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_JointTorqueUnit.ValueType], builtins.type
111
- ):
112
- DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
113
- NEWTON_METERS: _JointTorqueUnit.ValueType # 0
114
-
115
- class JointTorqueUnit(_JointTorqueUnit, metaclass=_JointTorqueUnitEnumTypeWrapper):
116
- """-------------- *
117
- Joint Torques *
118
- --------------
119
- """
120
-
121
- NEWTON_METERS: JointTorqueUnit.ValueType # 0
122
- global___JointTorqueUnit = JointTorqueUnit
123
-
124
- @typing.final
125
- class JointPositionValue(google.protobuf.message.Message):
126
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
127
-
128
- JOINT_NAME_FIELD_NUMBER: builtins.int
129
- VALUE_FIELD_NUMBER: builtins.int
130
- UNIT_FIELD_NUMBER: builtins.int
131
- joint_name: builtins.str
132
- value: builtins.float
133
- unit: global___JointPositionUnit.ValueType
134
- def __init__(
135
- self,
136
- *,
137
- joint_name: builtins.str = ...,
138
- value: builtins.float = ...,
139
- unit: global___JointPositionUnit.ValueType = ...,
140
- ) -> None: ...
141
- def ClearField(
142
- self, field_name: typing.Literal["joint_name", b"joint_name", "unit", b"unit", "value", b"value"]
143
- ) -> None: ...
144
-
145
- global___JointPositionValue = JointPositionValue
146
-
147
- @typing.final
148
- class JointPositionsSchema(google.protobuf.message.Message):
149
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
150
-
151
- UNIT_FIELD_NUMBER: builtins.int
152
- JOINT_NAMES_FIELD_NUMBER: builtins.int
153
- unit: global___JointPositionUnit.ValueType
154
- @property
155
- def joint_names(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
156
- def __init__(
157
- self,
158
- *,
159
- unit: global___JointPositionUnit.ValueType = ...,
160
- joint_names: collections.abc.Iterable[builtins.str] | None = ...,
161
- ) -> None: ...
162
- def ClearField(self, field_name: typing.Literal["joint_names", b"joint_names", "unit", b"unit"]) -> None: ...
163
-
164
- global___JointPositionsSchema = JointPositionsSchema
165
-
166
- @typing.final
167
- class JointPositionsValue(google.protobuf.message.Message):
168
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
169
-
170
- VALUES_FIELD_NUMBER: builtins.int
171
- @property
172
- def values(
173
- self,
174
- ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___JointPositionValue]: ...
175
- def __init__(
176
- self,
177
- *,
178
- values: collections.abc.Iterable[global___JointPositionValue] | None = ...,
179
- ) -> None: ...
180
- def ClearField(self, field_name: typing.Literal["values", b"values"]) -> None: ...
181
-
182
- global___JointPositionsValue = JointPositionsValue
183
-
184
- @typing.final
185
- class JointVelocityValue(google.protobuf.message.Message):
186
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
187
-
188
- JOINT_NAME_FIELD_NUMBER: builtins.int
189
- VALUE_FIELD_NUMBER: builtins.int
190
- UNIT_FIELD_NUMBER: builtins.int
191
- joint_name: builtins.str
192
- value: builtins.float
193
- unit: global___JointVelocityUnit.ValueType
194
- def __init__(
195
- self,
196
- *,
197
- joint_name: builtins.str = ...,
198
- value: builtins.float = ...,
199
- unit: global___JointVelocityUnit.ValueType = ...,
200
- ) -> None: ...
201
- def ClearField(
202
- self, field_name: typing.Literal["joint_name", b"joint_name", "unit", b"unit", "value", b"value"]
203
- ) -> None: ...
204
-
205
- global___JointVelocityValue = JointVelocityValue
206
-
207
- @typing.final
208
- class JointVelocitiesSchema(google.protobuf.message.Message):
209
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
210
-
211
- UNIT_FIELD_NUMBER: builtins.int
212
- JOINT_NAMES_FIELD_NUMBER: builtins.int
213
- unit: global___JointVelocityUnit.ValueType
214
- @property
215
- def joint_names(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
216
- def __init__(
217
- self,
218
- *,
219
- unit: global___JointVelocityUnit.ValueType = ...,
220
- joint_names: collections.abc.Iterable[builtins.str] | None = ...,
221
- ) -> None: ...
222
- def ClearField(self, field_name: typing.Literal["joint_names", b"joint_names", "unit", b"unit"]) -> None: ...
223
-
224
- global___JointVelocitiesSchema = JointVelocitiesSchema
225
-
226
- @typing.final
227
- class JointVelocitiesValue(google.protobuf.message.Message):
228
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
229
-
230
- VALUES_FIELD_NUMBER: builtins.int
231
- @property
232
- def values(
233
- self,
234
- ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___JointVelocityValue]: ...
235
- def __init__(
236
- self,
237
- *,
238
- values: collections.abc.Iterable[global___JointVelocityValue] | None = ...,
239
- ) -> None: ...
240
- def ClearField(self, field_name: typing.Literal["values", b"values"]) -> None: ...
241
-
242
- global___JointVelocitiesValue = JointVelocitiesValue
243
-
244
- @typing.final
245
- class JointTorqueValue(google.protobuf.message.Message):
246
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
247
-
248
- JOINT_NAME_FIELD_NUMBER: builtins.int
249
- VALUE_FIELD_NUMBER: builtins.int
250
- UNIT_FIELD_NUMBER: builtins.int
251
- joint_name: builtins.str
252
- value: builtins.float
253
- unit: global___JointTorqueUnit.ValueType
254
- def __init__(
255
- self,
256
- *,
257
- joint_name: builtins.str = ...,
258
- value: builtins.float = ...,
259
- unit: global___JointTorqueUnit.ValueType = ...,
260
- ) -> None: ...
261
- def ClearField(
262
- self, field_name: typing.Literal["joint_name", b"joint_name", "unit", b"unit", "value", b"value"]
263
- ) -> None: ...
264
-
265
- global___JointTorqueValue = JointTorqueValue
266
-
267
- @typing.final
268
- class JointTorquesSchema(google.protobuf.message.Message):
269
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
270
-
271
- UNIT_FIELD_NUMBER: builtins.int
272
- JOINT_NAMES_FIELD_NUMBER: builtins.int
273
- unit: global___JointTorqueUnit.ValueType
274
- @property
275
- def joint_names(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
276
- def __init__(
277
- self,
278
- *,
279
- unit: global___JointTorqueUnit.ValueType = ...,
280
- joint_names: collections.abc.Iterable[builtins.str] | None = ...,
281
- ) -> None: ...
282
- def ClearField(self, field_name: typing.Literal["joint_names", b"joint_names", "unit", b"unit"]) -> None: ...
283
-
284
- global___JointTorquesSchema = JointTorquesSchema
285
-
286
- @typing.final
287
- class JointTorquesValue(google.protobuf.message.Message):
288
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
289
-
290
- VALUES_FIELD_NUMBER: builtins.int
291
- @property
292
- def values(
293
- self,
294
- ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___JointTorqueValue]: ...
295
- def __init__(
296
- self,
297
- *,
298
- values: collections.abc.Iterable[global___JointTorqueValue] | None = ...,
299
- ) -> None: ...
300
- def ClearField(self, field_name: typing.Literal["values", b"values"]) -> None: ...
301
-
302
- global___JointTorquesValue = JointTorquesValue
303
-
304
- @typing.final
305
- class JointCommandsSchema(google.protobuf.message.Message):
306
- """---------------- *
307
- Actuator Command *
308
- ----------------
309
- """
310
-
311
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
312
-
313
- JOINT_NAMES_FIELD_NUMBER: builtins.int
314
- TORQUE_UNIT_FIELD_NUMBER: builtins.int
315
- VELOCITY_UNIT_FIELD_NUMBER: builtins.int
316
- POSITION_UNIT_FIELD_NUMBER: builtins.int
317
- torque_unit: global___JointTorqueUnit.ValueType
318
- velocity_unit: global___JointVelocityUnit.ValueType
319
- position_unit: global___JointPositionUnit.ValueType
320
- @property
321
- def joint_names(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
322
- def __init__(
323
- self,
324
- *,
325
- joint_names: collections.abc.Iterable[builtins.str] | None = ...,
326
- torque_unit: global___JointTorqueUnit.ValueType = ...,
327
- velocity_unit: global___JointVelocityUnit.ValueType = ...,
328
- position_unit: global___JointPositionUnit.ValueType = ...,
329
- ) -> None: ...
330
- def ClearField(
331
- self,
332
- field_name: typing.Literal[
333
- "joint_names",
334
- b"joint_names",
335
- "position_unit",
336
- b"position_unit",
337
- "torque_unit",
338
- b"torque_unit",
339
- "velocity_unit",
340
- b"velocity_unit",
341
- ],
342
- ) -> None: ...
343
-
344
- global___JointCommandsSchema = JointCommandsSchema
345
-
346
- @typing.final
347
- class JointCommandValue(google.protobuf.message.Message):
348
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
349
-
350
- JOINT_NAME_FIELD_NUMBER: builtins.int
351
- TORQUE_FIELD_NUMBER: builtins.int
352
- VELOCITY_FIELD_NUMBER: builtins.int
353
- POSITION_FIELD_NUMBER: builtins.int
354
- KP_FIELD_NUMBER: builtins.int
355
- KD_FIELD_NUMBER: builtins.int
356
- TORQUE_UNIT_FIELD_NUMBER: builtins.int
357
- VELOCITY_UNIT_FIELD_NUMBER: builtins.int
358
- POSITION_UNIT_FIELD_NUMBER: builtins.int
359
- joint_name: builtins.str
360
- torque: builtins.float
361
- velocity: builtins.float
362
- position: builtins.float
363
- kp: builtins.float
364
- kd: builtins.float
365
- torque_unit: global___JointTorqueUnit.ValueType
366
- velocity_unit: global___JointVelocityUnit.ValueType
367
- position_unit: global___JointPositionUnit.ValueType
368
- def __init__(
369
- self,
370
- *,
371
- joint_name: builtins.str = ...,
372
- torque: builtins.float = ...,
373
- velocity: builtins.float = ...,
374
- position: builtins.float = ...,
375
- kp: builtins.float = ...,
376
- kd: builtins.float = ...,
377
- torque_unit: global___JointTorqueUnit.ValueType = ...,
378
- velocity_unit: global___JointVelocityUnit.ValueType = ...,
379
- position_unit: global___JointPositionUnit.ValueType = ...,
380
- ) -> None: ...
381
- def ClearField(
382
- self,
383
- field_name: typing.Literal[
384
- "joint_name",
385
- b"joint_name",
386
- "kd",
387
- b"kd",
388
- "kp",
389
- b"kp",
390
- "position",
391
- b"position",
392
- "position_unit",
393
- b"position_unit",
394
- "torque",
395
- b"torque",
396
- "torque_unit",
397
- b"torque_unit",
398
- "velocity",
399
- b"velocity",
400
- "velocity_unit",
401
- b"velocity_unit",
402
- ],
403
- ) -> None: ...
404
-
405
- global___JointCommandValue = JointCommandValue
406
-
407
- @typing.final
408
- class JointCommandsValue(google.protobuf.message.Message):
409
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
410
-
411
- VALUES_FIELD_NUMBER: builtins.int
412
- @property
413
- def values(
414
- self,
415
- ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___JointCommandValue]: ...
416
- def __init__(
417
- self,
418
- *,
419
- values: collections.abc.Iterable[global___JointCommandValue] | None = ...,
420
- ) -> None: ...
421
- def ClearField(self, field_name: typing.Literal["values", b"values"]) -> None: ...
422
-
423
- global___JointCommandsValue = JointCommandsValue
424
-
425
- @typing.final
426
- class CameraFrameSchema(google.protobuf.message.Message):
427
- """------------ *
428
- Camera Frame *
429
- ------------
430
- """
431
-
432
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
433
-
434
- WIDTH_FIELD_NUMBER: builtins.int
435
- HEIGHT_FIELD_NUMBER: builtins.int
436
- CHANNELS_FIELD_NUMBER: builtins.int
437
- width: builtins.int
438
- height: builtins.int
439
- channels: builtins.int
440
- def __init__(
441
- self,
442
- *,
443
- width: builtins.int = ...,
444
- height: builtins.int = ...,
445
- channels: builtins.int = ...,
446
- ) -> None: ...
447
- def ClearField(
448
- self, field_name: typing.Literal["channels", b"channels", "height", b"height", "width", b"width"]
449
- ) -> None: ...
450
-
451
- global___CameraFrameSchema = CameraFrameSchema
452
-
453
- @typing.final
454
- class CameraFrameValue(google.protobuf.message.Message):
455
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
456
-
457
- DATA_FIELD_NUMBER: builtins.int
458
- data: builtins.bytes
459
- def __init__(
460
- self,
461
- *,
462
- data: builtins.bytes = ...,
463
- ) -> None: ...
464
- def ClearField(self, field_name: typing.Literal["data", b"data"]) -> None: ...
465
-
466
- global___CameraFrameValue = CameraFrameValue
467
-
468
- @typing.final
469
- class AudioFrameSchema(google.protobuf.message.Message):
470
- """----------- *
471
- Audio Frame *
472
- -----------
473
- """
474
-
475
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
476
-
477
- CHANNELS_FIELD_NUMBER: builtins.int
478
- SAMPLE_RATE_FIELD_NUMBER: builtins.int
479
- DTYPE_FIELD_NUMBER: builtins.int
480
- channels: builtins.int
481
- sample_rate: builtins.int
482
- dtype: global___DType.ValueType
483
- def __init__(
484
- self,
485
- *,
486
- channels: builtins.int = ...,
487
- sample_rate: builtins.int = ...,
488
- dtype: global___DType.ValueType = ...,
489
- ) -> None: ...
490
- def ClearField(
491
- self, field_name: typing.Literal["channels", b"channels", "dtype", b"dtype", "sample_rate", b"sample_rate"]
492
- ) -> None: ...
493
-
494
- global___AudioFrameSchema = AudioFrameSchema
495
-
496
- @typing.final
497
- class AudioFrameValue(google.protobuf.message.Message):
498
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
499
-
500
- DATA_FIELD_NUMBER: builtins.int
501
- data: builtins.bytes
502
- def __init__(
503
- self,
504
- *,
505
- data: builtins.bytes = ...,
506
- ) -> None: ...
507
- def ClearField(self, field_name: typing.Literal["data", b"data"]) -> None: ...
508
-
509
- global___AudioFrameValue = AudioFrameValue
510
-
511
- @typing.final
512
- class ImuAccelerometerValue(google.protobuf.message.Message):
513
- """--- *
514
- IMU *
515
- ---
516
- """
517
-
518
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
519
-
520
- X_FIELD_NUMBER: builtins.int
521
- Y_FIELD_NUMBER: builtins.int
522
- Z_FIELD_NUMBER: builtins.int
523
- x: builtins.float
524
- y: builtins.float
525
- z: builtins.float
526
- def __init__(
527
- self,
528
- *,
529
- x: builtins.float = ...,
530
- y: builtins.float = ...,
531
- z: builtins.float = ...,
532
- ) -> None: ...
533
- def ClearField(self, field_name: typing.Literal["x", b"x", "y", b"y", "z", b"z"]) -> None: ...
534
-
535
- global___ImuAccelerometerValue = ImuAccelerometerValue
536
-
537
- @typing.final
538
- class ImuGyroscopeValue(google.protobuf.message.Message):
539
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
540
-
541
- X_FIELD_NUMBER: builtins.int
542
- Y_FIELD_NUMBER: builtins.int
543
- Z_FIELD_NUMBER: builtins.int
544
- x: builtins.float
545
- y: builtins.float
546
- z: builtins.float
547
- def __init__(
548
- self,
549
- *,
550
- x: builtins.float = ...,
551
- y: builtins.float = ...,
552
- z: builtins.float = ...,
553
- ) -> None: ...
554
- def ClearField(self, field_name: typing.Literal["x", b"x", "y", b"y", "z", b"z"]) -> None: ...
555
-
556
- global___ImuGyroscopeValue = ImuGyroscopeValue
557
-
558
- @typing.final
559
- class ImuMagnetometerValue(google.protobuf.message.Message):
560
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
561
-
562
- X_FIELD_NUMBER: builtins.int
563
- Y_FIELD_NUMBER: builtins.int
564
- Z_FIELD_NUMBER: builtins.int
565
- x: builtins.float
566
- y: builtins.float
567
- z: builtins.float
568
- def __init__(
569
- self,
570
- *,
571
- x: builtins.float = ...,
572
- y: builtins.float = ...,
573
- z: builtins.float = ...,
574
- ) -> None: ...
575
- def ClearField(self, field_name: typing.Literal["x", b"x", "y", b"y", "z", b"z"]) -> None: ...
576
-
577
- global___ImuMagnetometerValue = ImuMagnetometerValue
578
-
579
- @typing.final
580
- class ImuSchema(google.protobuf.message.Message):
581
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
582
-
583
- USE_ACCELEROMETER_FIELD_NUMBER: builtins.int
584
- USE_GYROSCOPE_FIELD_NUMBER: builtins.int
585
- USE_MAGNETOMETER_FIELD_NUMBER: builtins.int
586
- use_accelerometer: builtins.bool
587
- use_gyroscope: builtins.bool
588
- use_magnetometer: builtins.bool
589
- def __init__(
590
- self,
591
- *,
592
- use_accelerometer: builtins.bool = ...,
593
- use_gyroscope: builtins.bool = ...,
594
- use_magnetometer: builtins.bool = ...,
595
- ) -> None: ...
596
- def ClearField(
597
- self,
598
- field_name: typing.Literal[
599
- "use_accelerometer",
600
- b"use_accelerometer",
601
- "use_gyroscope",
602
- b"use_gyroscope",
603
- "use_magnetometer",
604
- b"use_magnetometer",
605
- ],
606
- ) -> None: ...
607
-
608
- global___ImuSchema = ImuSchema
609
-
610
- @typing.final
611
- class ImuValue(google.protobuf.message.Message):
612
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
613
-
614
- LINEAR_ACCELERATION_FIELD_NUMBER: builtins.int
615
- ANGULAR_VELOCITY_FIELD_NUMBER: builtins.int
616
- MAGNETIC_FIELD_FIELD_NUMBER: builtins.int
617
- @property
618
- def linear_acceleration(self) -> global___ImuAccelerometerValue: ...
619
- @property
620
- def angular_velocity(self) -> global___ImuGyroscopeValue: ...
621
- @property
622
- def magnetic_field(self) -> global___ImuMagnetometerValue: ...
623
- def __init__(
624
- self,
625
- *,
626
- linear_acceleration: global___ImuAccelerometerValue | None = ...,
627
- angular_velocity: global___ImuGyroscopeValue | None = ...,
628
- magnetic_field: global___ImuMagnetometerValue | None = ...,
629
- ) -> None: ...
630
- def HasField(
631
- self,
632
- field_name: typing.Literal[
633
- "angular_velocity",
634
- b"angular_velocity",
635
- "linear_acceleration",
636
- b"linear_acceleration",
637
- "magnetic_field",
638
- b"magnetic_field",
639
- ],
640
- ) -> builtins.bool: ...
641
- def ClearField(
642
- self,
643
- field_name: typing.Literal[
644
- "angular_velocity",
645
- b"angular_velocity",
646
- "linear_acceleration",
647
- b"linear_acceleration",
648
- "magnetic_field",
649
- b"magnetic_field",
650
- ],
651
- ) -> None: ...
652
-
653
- global___ImuValue = ImuValue
654
-
655
- @typing.final
656
- class TimestampSchema(google.protobuf.message.Message):
657
- """--------- *
658
- Timestamp *
659
- ---------
660
- """
661
-
662
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
663
-
664
- START_SECONDS_FIELD_NUMBER: builtins.int
665
- START_NANOS_FIELD_NUMBER: builtins.int
666
- start_seconds: builtins.int
667
- start_nanos: builtins.int
668
- def __init__(
669
- self,
670
- *,
671
- start_seconds: builtins.int = ...,
672
- start_nanos: builtins.int = ...,
673
- ) -> None: ...
674
- def ClearField(
675
- self, field_name: typing.Literal["start_nanos", b"start_nanos", "start_seconds", b"start_seconds"]
676
- ) -> None: ...
677
-
678
- global___TimestampSchema = TimestampSchema
679
-
680
- @typing.final
681
- class TimestampValue(google.protobuf.message.Message):
682
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
683
-
684
- SECONDS_FIELD_NUMBER: builtins.int
685
- NANOS_FIELD_NUMBER: builtins.int
686
- seconds: builtins.int
687
- nanos: builtins.int
688
- def __init__(
689
- self,
690
- *,
691
- seconds: builtins.int = ...,
692
- nanos: builtins.int = ...,
693
- ) -> None: ...
694
- def ClearField(self, field_name: typing.Literal["nanos", b"nanos", "seconds", b"seconds"]) -> None: ...
695
-
696
- global___TimestampValue = TimestampValue
697
-
698
- @typing.final
699
- class VectorCommandSchema(google.protobuf.message.Message):
700
- """------------------- *
701
- XY Position Command *
702
- -------------------
703
- """
704
-
705
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
706
-
707
- DIMENSIONS_FIELD_NUMBER: builtins.int
708
- dimensions: builtins.int
709
- def __init__(
710
- self,
711
- *,
712
- dimensions: builtins.int = ...,
713
- ) -> None: ...
714
- def ClearField(self, field_name: typing.Literal["dimensions", b"dimensions"]) -> None: ...
715
-
716
- global___VectorCommandSchema = VectorCommandSchema
717
-
718
- @typing.final
719
- class VectorCommandValue(google.protobuf.message.Message):
720
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
721
-
722
- VALUES_FIELD_NUMBER: builtins.int
723
- @property
724
- def values(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.float]: ...
725
- def __init__(
726
- self,
727
- *,
728
- values: collections.abc.Iterable[builtins.float] | None = ...,
729
- ) -> None: ...
730
- def ClearField(self, field_name: typing.Literal["values", b"values"]) -> None: ...
731
-
732
- global___VectorCommandValue = VectorCommandValue
733
-
734
- @typing.final
735
- class StateTensorSchema(google.protobuf.message.Message):
736
- """------------ *
737
- State Tensor *
738
- ------------
739
- """
740
-
741
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
742
-
743
- SHAPE_FIELD_NUMBER: builtins.int
744
- DTYPE_FIELD_NUMBER: builtins.int
745
- dtype: global___DType.ValueType
746
- @property
747
- def shape(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ...
748
- def __init__(
749
- self,
750
- *,
751
- shape: collections.abc.Iterable[builtins.int] | None = ...,
752
- dtype: global___DType.ValueType = ...,
753
- ) -> None: ...
754
- def ClearField(self, field_name: typing.Literal["dtype", b"dtype", "shape", b"shape"]) -> None: ...
755
-
756
- global___StateTensorSchema = StateTensorSchema
757
-
758
- @typing.final
759
- class StateTensorValue(google.protobuf.message.Message):
760
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
761
-
762
- DATA_FIELD_NUMBER: builtins.int
763
- data: builtins.bytes
764
- def __init__(
765
- self,
766
- *,
767
- data: builtins.bytes = ...,
768
- ) -> None: ...
769
- def ClearField(self, field_name: typing.Literal["data", b"data"]) -> None: ...
770
-
771
- global___StateTensorValue = StateTensorValue
772
-
773
- @typing.final
774
- class Value(google.protobuf.message.Message):
775
- """----- *
776
- Value *
777
- -----
778
- """
779
-
780
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
781
-
782
- VALUE_NAME_FIELD_NUMBER: builtins.int
783
- JOINT_POSITIONS_FIELD_NUMBER: builtins.int
784
- JOINT_VELOCITIES_FIELD_NUMBER: builtins.int
785
- JOINT_TORQUES_FIELD_NUMBER: builtins.int
786
- JOINT_COMMANDS_FIELD_NUMBER: builtins.int
787
- CAMERA_FRAME_FIELD_NUMBER: builtins.int
788
- AUDIO_FRAME_FIELD_NUMBER: builtins.int
789
- IMU_FIELD_NUMBER: builtins.int
790
- TIMESTAMP_FIELD_NUMBER: builtins.int
791
- VECTOR_COMMAND_FIELD_NUMBER: builtins.int
792
- STATE_TENSOR_FIELD_NUMBER: builtins.int
793
- value_name: builtins.str
794
- @property
795
- def joint_positions(self) -> global___JointPositionsValue: ...
796
- @property
797
- def joint_velocities(self) -> global___JointVelocitiesValue: ...
798
- @property
799
- def joint_torques(self) -> global___JointTorquesValue: ...
800
- @property
801
- def joint_commands(self) -> global___JointCommandsValue: ...
802
- @property
803
- def camera_frame(self) -> global___CameraFrameValue: ...
804
- @property
805
- def audio_frame(self) -> global___AudioFrameValue: ...
806
- @property
807
- def imu(self) -> global___ImuValue: ...
808
- @property
809
- def timestamp(self) -> global___TimestampValue: ...
810
- @property
811
- def vector_command(self) -> global___VectorCommandValue: ...
812
- @property
813
- def state_tensor(self) -> global___StateTensorValue: ...
814
- def __init__(
815
- self,
816
- *,
817
- value_name: builtins.str = ...,
818
- joint_positions: global___JointPositionsValue | None = ...,
819
- joint_velocities: global___JointVelocitiesValue | None = ...,
820
- joint_torques: global___JointTorquesValue | None = ...,
821
- joint_commands: global___JointCommandsValue | None = ...,
822
- camera_frame: global___CameraFrameValue | None = ...,
823
- audio_frame: global___AudioFrameValue | None = ...,
824
- imu: global___ImuValue | None = ...,
825
- timestamp: global___TimestampValue | None = ...,
826
- vector_command: global___VectorCommandValue | None = ...,
827
- state_tensor: global___StateTensorValue | None = ...,
828
- ) -> None: ...
829
- def HasField(
830
- self,
831
- field_name: typing.Literal[
832
- "audio_frame",
833
- b"audio_frame",
834
- "camera_frame",
835
- b"camera_frame",
836
- "imu",
837
- b"imu",
838
- "joint_commands",
839
- b"joint_commands",
840
- "joint_positions",
841
- b"joint_positions",
842
- "joint_torques",
843
- b"joint_torques",
844
- "joint_velocities",
845
- b"joint_velocities",
846
- "state_tensor",
847
- b"state_tensor",
848
- "timestamp",
849
- b"timestamp",
850
- "value",
851
- b"value",
852
- "vector_command",
853
- b"vector_command",
854
- ],
855
- ) -> builtins.bool: ...
856
- def ClearField(
857
- self,
858
- field_name: typing.Literal[
859
- "audio_frame",
860
- b"audio_frame",
861
- "camera_frame",
862
- b"camera_frame",
863
- "imu",
864
- b"imu",
865
- "joint_commands",
866
- b"joint_commands",
867
- "joint_positions",
868
- b"joint_positions",
869
- "joint_torques",
870
- b"joint_torques",
871
- "joint_velocities",
872
- b"joint_velocities",
873
- "state_tensor",
874
- b"state_tensor",
875
- "timestamp",
876
- b"timestamp",
877
- "value",
878
- b"value",
879
- "value_name",
880
- b"value_name",
881
- "vector_command",
882
- b"vector_command",
883
- ],
884
- ) -> None: ...
885
- def WhichOneof(
886
- self, oneof_group: typing.Literal["value", b"value"]
887
- ) -> (
888
- typing.Literal[
889
- "joint_positions",
890
- "joint_velocities",
891
- "joint_torques",
892
- "joint_commands",
893
- "camera_frame",
894
- "audio_frame",
895
- "imu",
896
- "timestamp",
897
- "vector_command",
898
- "state_tensor",
899
- ]
900
- | None
901
- ): ...
902
-
903
- global___Value = Value
904
-
905
- @typing.final
906
- class ValueSchema(google.protobuf.message.Message):
907
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
908
-
909
- VALUE_NAME_FIELD_NUMBER: builtins.int
910
- JOINT_POSITIONS_FIELD_NUMBER: builtins.int
911
- JOINT_VELOCITIES_FIELD_NUMBER: builtins.int
912
- JOINT_TORQUES_FIELD_NUMBER: builtins.int
913
- JOINT_COMMANDS_FIELD_NUMBER: builtins.int
914
- CAMERA_FRAME_FIELD_NUMBER: builtins.int
915
- AUDIO_FRAME_FIELD_NUMBER: builtins.int
916
- IMU_FIELD_NUMBER: builtins.int
917
- TIMESTAMP_FIELD_NUMBER: builtins.int
918
- VECTOR_COMMAND_FIELD_NUMBER: builtins.int
919
- STATE_TENSOR_FIELD_NUMBER: builtins.int
920
- value_name: builtins.str
921
- @property
922
- def joint_positions(self) -> global___JointPositionsSchema: ...
923
- @property
924
- def joint_velocities(self) -> global___JointVelocitiesSchema: ...
925
- @property
926
- def joint_torques(self) -> global___JointTorquesSchema: ...
927
- @property
928
- def joint_commands(self) -> global___JointCommandsSchema: ...
929
- @property
930
- def camera_frame(self) -> global___CameraFrameSchema: ...
931
- @property
932
- def audio_frame(self) -> global___AudioFrameSchema: ...
933
- @property
934
- def imu(self) -> global___ImuSchema: ...
935
- @property
936
- def timestamp(self) -> global___TimestampSchema: ...
937
- @property
938
- def vector_command(self) -> global___VectorCommandSchema: ...
939
- @property
940
- def state_tensor(self) -> global___StateTensorSchema: ...
941
- def __init__(
942
- self,
943
- *,
944
- value_name: builtins.str = ...,
945
- joint_positions: global___JointPositionsSchema | None = ...,
946
- joint_velocities: global___JointVelocitiesSchema | None = ...,
947
- joint_torques: global___JointTorquesSchema | None = ...,
948
- joint_commands: global___JointCommandsSchema | None = ...,
949
- camera_frame: global___CameraFrameSchema | None = ...,
950
- audio_frame: global___AudioFrameSchema | None = ...,
951
- imu: global___ImuSchema | None = ...,
952
- timestamp: global___TimestampSchema | None = ...,
953
- vector_command: global___VectorCommandSchema | None = ...,
954
- state_tensor: global___StateTensorSchema | None = ...,
955
- ) -> None: ...
956
- def HasField(
957
- self,
958
- field_name: typing.Literal[
959
- "audio_frame",
960
- b"audio_frame",
961
- "camera_frame",
962
- b"camera_frame",
963
- "imu",
964
- b"imu",
965
- "joint_commands",
966
- b"joint_commands",
967
- "joint_positions",
968
- b"joint_positions",
969
- "joint_torques",
970
- b"joint_torques",
971
- "joint_velocities",
972
- b"joint_velocities",
973
- "state_tensor",
974
- b"state_tensor",
975
- "timestamp",
976
- b"timestamp",
977
- "value_type",
978
- b"value_type",
979
- "vector_command",
980
- b"vector_command",
981
- ],
982
- ) -> builtins.bool: ...
983
- def ClearField(
984
- self,
985
- field_name: typing.Literal[
986
- "audio_frame",
987
- b"audio_frame",
988
- "camera_frame",
989
- b"camera_frame",
990
- "imu",
991
- b"imu",
992
- "joint_commands",
993
- b"joint_commands",
994
- "joint_positions",
995
- b"joint_positions",
996
- "joint_torques",
997
- b"joint_torques",
998
- "joint_velocities",
999
- b"joint_velocities",
1000
- "state_tensor",
1001
- b"state_tensor",
1002
- "timestamp",
1003
- b"timestamp",
1004
- "value_name",
1005
- b"value_name",
1006
- "value_type",
1007
- b"value_type",
1008
- "vector_command",
1009
- b"vector_command",
1010
- ],
1011
- ) -> None: ...
1012
- def WhichOneof(
1013
- self, oneof_group: typing.Literal["value_type", b"value_type"]
1014
- ) -> (
1015
- typing.Literal[
1016
- "joint_positions",
1017
- "joint_velocities",
1018
- "joint_torques",
1019
- "joint_commands",
1020
- "camera_frame",
1021
- "audio_frame",
1022
- "imu",
1023
- "timestamp",
1024
- "vector_command",
1025
- "state_tensor",
1026
- ]
1027
- | None
1028
- ): ...
1029
-
1030
- global___ValueSchema = ValueSchema
1031
-
1032
- @typing.final
1033
- class IOSchema(google.protobuf.message.Message):
1034
- """---------------- *
1035
- Inputs / Outputs *
1036
- ----------------
1037
- """
1038
-
1039
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
1040
-
1041
- VALUES_FIELD_NUMBER: builtins.int
1042
- @property
1043
- def values(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ValueSchema]: ...
1044
- def __init__(
1045
- self,
1046
- *,
1047
- values: collections.abc.Iterable[global___ValueSchema] | None = ...,
1048
- ) -> None: ...
1049
- def ClearField(self, field_name: typing.Literal["values", b"values"]) -> None: ...
1050
-
1051
- global___IOSchema = IOSchema
1052
-
1053
- @typing.final
1054
- class IO(google.protobuf.message.Message):
1055
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
1056
-
1057
- VALUES_FIELD_NUMBER: builtins.int
1058
- @property
1059
- def values(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Value]: ...
1060
- def __init__(
1061
- self,
1062
- *,
1063
- values: collections.abc.Iterable[global___Value] | None = ...,
1064
- ) -> None: ...
1065
- def ClearField(self, field_name: typing.Literal["values", b"values"]) -> None: ...
1066
-
1067
- global___IO = IO
1068
-
1069
- @typing.final
1070
- class ModelSchema(google.protobuf.message.Message):
1071
- """----- *
1072
- Model *
1073
- -----
1074
- """
1075
-
1076
- DESCRIPTOR: google.protobuf.descriptor.Descriptor
1077
-
1078
- INPUT_SCHEMA_FIELD_NUMBER: builtins.int
1079
- OUTPUT_SCHEMA_FIELD_NUMBER: builtins.int
1080
- @property
1081
- def input_schema(self) -> global___IOSchema: ...
1082
- @property
1083
- def output_schema(self) -> global___IOSchema: ...
1084
- def __init__(
1085
- self,
1086
- *,
1087
- input_schema: global___IOSchema | None = ...,
1088
- output_schema: global___IOSchema | None = ...,
1089
- ) -> None: ...
1090
- def HasField(
1091
- self, field_name: typing.Literal["input_schema", b"input_schema", "output_schema", b"output_schema"]
1092
- ) -> builtins.bool: ...
1093
- def ClearField(
1094
- self, field_name: typing.Literal["input_schema", b"input_schema", "output_schema", b"output_schema"]
1095
- ) -> None: ...
1096
-
1097
- global___ModelSchema = ModelSchema