kinfer 0.0.5__tar.gz → 0.3.1__tar.gz

kinfer-0.3.1/Cargo.toml

@@ -0,0 +1,18 @@
+[workspace]
+
+members = [
+    "kinfer/rust",
+    "kinfer/rust_bindings",
+]
+resolver = "2"
+
+[workspace.package]
+
+version = "0.3.1"
+authors = ["Wesley Maa <wesley@kscale.dev>", "Benjamin Bolte <ben@kscale.dev>"]
+edition = "2021"
+description = "K-Scale Inference Library"
+license = "MIT"
+repository = "https://github.com/kscalelabs/kinfer"
+documentation = "https://docs.kscale.dev"
+readme = "README.md"

kinfer-0.3.1/MANIFEST.in

@@ -0,0 +1,2 @@
+include Cargo.toml
+recursive-include kinfer/ *.rs *.pyi *.so *.py *.txt *.md *.toml py.typed

kinfer-0.3.1/PKG-INFO

@@ -0,0 +1,57 @@
+Metadata-Version: 2.1
+Name: kinfer
+Version: 0.3.1
+Summary: Tool to make it easier to run a model on a real robot
+Home-page: https://github.com/kscalelabs/kinfer.git
+Author: K-Scale Labs
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: torch
+Requires-Dist: onnx
+Requires-Dist: onnxruntime
+Requires-Dist: protobuf
+Provides-Extra: dev
+Requires-Dist: black; extra == "dev"
+Requires-Dist: darglint; extra == "dev"
+Requires-Dist: mypy; extra == "dev"
+Requires-Dist: mypy-protobuf; extra == "dev"
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: ruff; extra == "dev"
+
+# kinfer
+
+This package is designed to support exporting and running inference on PyTorch models.
+
+## Installation
+
+```bash
+pip install kinfer
+```
+
+### ONNX Runtime
+
+You can install the latest version of ONNX Runtime on Mac with:
+
+```bash
+brew install onnxruntime
+```
+
+You may need to add the binary to your DYLD_LIBRARY_PATH:
+
+```bash
+$ brew ls onnxruntime
+/opt/homebrew/Cellar/onnxruntime/1.20.1/include/onnxruntime/ (11 files)
+/opt/homebrew/Cellar/onnxruntime/1.20.1/lib/libonnxruntime.1.20.1.dylib  # <-- This is the binary
+/opt/homebrew/Cellar/onnxruntime/1.20.1/lib/cmake/ (4 files)
+/opt/homebrew/Cellar/onnxruntime/1.20.1/lib/pkgconfig/libonnxruntime.pc
+/opt/homebrew/Cellar/onnxruntime/1.20.1/lib/libonnxruntime.dylib
+/opt/homebrew/Cellar/onnxruntime/1.20.1/sbom.spdx.json
+$ export DYLD_LIBRARY_PATH=/opt/homebrew/Cellar/onnxruntime/1.20.1/lib:$DYLD_LIBRARY_PATH
+```
+
+### Considerations for Exporting PyTorch Models
+
+Don't use common names for the inputs to your forward pass. E.g. `input`, `output`, `state`, `state_tensor`, `buffer`, etc.
+
+This is because ONNX has internal names for the model and if there's a conflict, the inputs will have a .1, .2, etc. suffix which makes it really hard to figure out what value_name to pass into your kinfer io values. 

kinfer-0.3.1/README.md

@@ -0,0 +1,36 @@
+# kinfer
+
+This package is designed to support exporting and running inference on PyTorch models.
+
+## Installation
+
+```bash
+pip install kinfer
+```
+
+### ONNX Runtime
+
+You can install the latest version of ONNX Runtime on Mac with:
+
+```bash
+brew install onnxruntime
+```
+
+You may need to add the binary to your DYLD_LIBRARY_PATH:
+
+```bash
+$ brew ls onnxruntime
+/opt/homebrew/Cellar/onnxruntime/1.20.1/include/onnxruntime/ (11 files)
+/opt/homebrew/Cellar/onnxruntime/1.20.1/lib/libonnxruntime.1.20.1.dylib  # <-- This is the binary
+/opt/homebrew/Cellar/onnxruntime/1.20.1/lib/cmake/ (4 files)
+/opt/homebrew/Cellar/onnxruntime/1.20.1/lib/pkgconfig/libonnxruntime.pc
+/opt/homebrew/Cellar/onnxruntime/1.20.1/lib/libonnxruntime.dylib
+/opt/homebrew/Cellar/onnxruntime/1.20.1/sbom.spdx.json
+$ export DYLD_LIBRARY_PATH=/opt/homebrew/Cellar/onnxruntime/1.20.1/lib:$DYLD_LIBRARY_PATH
+```
+
+### Considerations for Exporting PyTorch Models
+
+Don't use common names for the inputs to your forward pass. E.g. `input`, `output`, `state`, `state_tensor`, `buffer`, etc.
+
+This is because ONNX has internal names for the model and if there's a conflict, the inputs will have a .1, .2, etc. suffix which makes it really hard to figure out what value_name to pass into your kinfer io values. 

kinfer-0.3.1/kinfer/__init__.py

@@ -0,0 +1,6 @@
+"""Defines the kinfer API."""
+
+from . import export, inference
+from .rust_bindings import get_version
+
+__version__ = get_version()

kinfer-0.3.1/kinfer/export/pytorch.py

@@ -0,0 +1,128 @@
+"""PyTorch model export utilities."""
+
+import base64
+import inspect
+from io import BytesIO
+from typing import Sequence
+
+import onnx
+import onnxruntime as ort
+import torch
+from torch import Tensor
+
+from kinfer import proto as P
+from kinfer.serialize.pytorch import PyTorchMultiSerializer
+from kinfer.serialize.schema import get_dummy_io
+from kinfer.serialize.utils import check_names_match
+
+KINFER_METADATA_KEY = "kinfer_metadata"
+
+
+def _add_metadata_to_onnx(model_proto: onnx.ModelProto, schema: P.ModelSchema) -> onnx.ModelProto:
+    """Add metadata to ONNX model.
+
+    Args:
+        model_proto: ONNX model prototype
+        schema: Model schema to use for model export.
+
+    Returns:
+        ONNX model with added metadata
+    """
+    schema_bytes = schema.SerializeToString()
+
+    meta = model_proto.metadata_props.add()
+    meta.key = KINFER_METADATA_KEY
+    meta.value = base64.b64encode(schema_bytes).decode("utf-8")
+
+    return model_proto
+
+
+def export_model(model: torch.jit.ScriptModule, schema: P.ModelSchema) -> onnx.ModelProto:
+    """Export PyTorch model to ONNX format with metadata.
+
+    Args:
+        model: PyTorch model to export.
+        schema: Model schema to use for model export.
+
+    Returns:
+        ONNX inference session
+    """
+    # Matches each input name to the input values.
+    signature = inspect.signature(model.forward)
+    model_input_names = [
+        p.name for p in signature.parameters.values() if p.kind == inspect.Parameter.POSITIONAL_OR_KEYWORD
+    ]
+    if len(model_input_names) != len(schema.input_schema.values):
+        raise ValueError(f"Expected {len(model_input_names)} inputs, but schema has {len(schema.input_schema.values)}")
+    input_schema_names = [i.value_name for i in schema.input_schema.values]
+    output_schema_names = [o.value_name for o in schema.output_schema.values]
+
+    if model_input_names != input_schema_names:
+        raise ValueError(f"Expected input names {model_input_names} to match schema names {input_schema_names}")
+
+    input_serializer = PyTorchMultiSerializer(schema.input_schema)
+    output_serializer = PyTorchMultiSerializer(schema.output_schema)
+
+    input_dummy_values = get_dummy_io(schema.input_schema)
+    input_tensors = input_serializer.serialize_io(input_dummy_values, as_dict=True)
+
+    check_names_match("model_input_names", model_input_names, "input_schema", list(input_tensors.keys()))
+    input_tensor_list = [input_tensors[name] for name in model_input_names]
+
+    # Attempts to run the model with the dummy inputs.
+    try:
+        pred_output_tensors = model(*input_tensor_list)
+    except Exception as e:
+        signature = inspect.signature(model.forward)
+        model_input_names = [
+            p.name for p in signature.parameters.values() if p.kind == inspect.Parameter.POSITIONAL_OR_KEYWORD
+        ]
+
+        raise ValueError(
+            f"Failed to run model with dummy inputs; input names are {model_input_names} while "
+            f"input schema is {schema.input_schema}"
+        ) from e
+
+    # Attempts to parse the output tensors using the output schema.
+    if isinstance(pred_output_tensors, Tensor):
+        pred_output_tensors = (pred_output_tensors,)
+    if isinstance(pred_output_tensors, Sequence):
+        pred_output_tensors = output_serializer.assign_names(pred_output_tensors)
+    if not isinstance(pred_output_tensors, dict):
+        raise ValueError("Output tensors could not be converted to dictionary")
+    try:
+        pred_output_tensors = output_serializer.deserialize_io(pred_output_tensors)
+    except Exception as e:
+        raise ValueError("Failed to parse output tensors using output schema; are you sure it is correct?") from e
+
+    # Export model to buffer
+    buffer = BytesIO()
+    torch.onnx.export(
+        model=model,
+        f=buffer,  # type: ignore[arg-type]
+        kwargs=input_tensors,
+        input_names=input_schema_names,
+        output_names=output_schema_names,
+    )
+    buffer.seek(0)
+
+    # Loads the model from the buffer and adds metadata.
+    model_proto = onnx.load_model(buffer)
+    model_proto = _add_metadata_to_onnx(model_proto, schema)
+
+    return model_proto
+
+
+def get_model(model_proto: onnx.ModelProto) -> ort.InferenceSession:
+    """Converts a model proto to an inference session.
+
+    Args:
+        model_proto: ONNX model proto to convert to inference session.
+
+    Returns:
+        ONNX inference session
+    """
+    buffer = BytesIO()
+    onnx.save_model(model_proto, buffer)
+    buffer.seek(0)
+    return ort.InferenceSession(buffer.read())

kinfer-0.3.1/kinfer/inference/python.py

@@ -0,0 +1,92 @@
+"""ONNX model inference utilities for Python."""
+
+import base64
+from pathlib import Path
+
+import onnx
+import onnxruntime as ort
+
+from kinfer import proto as P
+from kinfer.export.pytorch import KINFER_METADATA_KEY
+from kinfer.serialize.numpy import NumpyMultiSerializer
+
+
+def _read_schema(model: onnx.ModelProto) -> P.ModelSchema:
+    for prop in model.metadata_props:
+        if prop.key == KINFER_METADATA_KEY:
+            try:
+                schema_bytes = base64.b64decode(prop.value)
+                schema = P.ModelSchema()
+                schema.ParseFromString(schema_bytes)
+                return schema
+            except Exception as e:
+                raise ValueError("Failed to parse kinfer_metadata value") from e
+        else:
+            raise ValueError(f"Found arbitrary metadata key {prop.key}")
+
+    raise ValueError(f"{KINFER_METADATA_KEY} not found in model metadata")
+
+
+class ONNXModel:
+    """Wrapper for ONNX model inference."""
+
+    def __init__(self: "ONNXModel", model_path: str | Path) -> None:
+        """Initialize ONNX model.
+
+        Args:
+            model_path: Path to ONNX model file
+        """
+        self.model_path = model_path
+
+        # Load model and create inference session
+        self.model = onnx.load(model_path)
+        self.session = ort.InferenceSession(model_path)
+        self._schema = _read_schema(self.model)
+
+        # Create serializers for input and output.
+        self._input_serializer = NumpyMultiSerializer(self._schema.input_schema)
+        self._output_serializer = NumpyMultiSerializer(self._schema.output_schema)
+
+    def __call__(self: "ONNXModel", inputs: P.IO) -> P.IO:
+        """Run inference on input data.
+
+        Args:
+            inputs: Input data, matching the input schema.
+
+        Returns:
+            Model outputs, matching the output schema.
+        """
+        inputs_np = self._input_serializer.serialize_io(inputs, as_dict=True)
+        outputs_np = self.session.run(None, inputs_np)
+        outputs = self._output_serializer.deserialize_io(outputs_np)
+        return outputs
+
+    @property
+    def input_schema(self: "ONNXModel") -> P.IOSchema:
+        """Get the input schema."""
+        return self._schema.input_schema
+
+    @property
+    def output_schema(self: "ONNXModel") -> P.IOSchema:
+        """Get the output schema."""
+        return self._schema.output_schema
+
+    @property
+    def schema_input_keys(self: "ONNXModel") -> list[str]:
+        """Get all value names from input schemas.
+
+        Returns:
+            List of value names from input schema.
+        """
+        input_names = [value.value_name for value in self._schema.input_schema.values]
+        return input_names
+
+    @property
+    def schema_output_keys(self: "ONNXModel") -> list[str]:
+        """Get all value names from output schemas.
+
+        Returns:
+            List of value names from output schema.
+        """
+        output_names = [value.value_name for value in self._schema.output_schema.values]
+        return output_names

kinfer-0.3.1/kinfer/proto/__init__.py

@@ -0,0 +1,40 @@
+"""Defines helper types for the protocol buffers."""
+
+from .kinfer_pb2 import (
+    IO,
+    AudioFrameSchema,
+    AudioFrameValue,
+    CameraFrameSchema,
+    CameraFrameValue,
+    DType,
+    ImuAccelerometerValue,
+    ImuGyroscopeValue,
+    ImuMagnetometerValue,
+    ImuSchema,
+    ImuValue,
+    IOSchema,
+    JointCommandsSchema,
+    JointCommandsValue,
+    JointCommandValue,
+    JointPositionsSchema,
+    JointPositionsValue,
+    JointPositionUnit,
+    JointPositionValue,
+    JointTorquesSchema,
+    JointTorquesValue,
+    JointTorqueUnit,
+    JointTorqueValue,
+    JointVelocitiesSchema,
+    JointVelocitiesValue,
+    JointVelocityUnit,
+    JointVelocityValue,
+    ModelSchema,
+    StateTensorSchema,
+    StateTensorValue,
+    TimestampSchema,
+    TimestampValue,
+    Value,
+    ValueSchema,
+    VectorCommandSchema,
+    VectorCommandValue,
+)

kinfer-0.3.1/kinfer/proto/kinfer_pb2.py

@@ -0,0 +1,103 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: kinfer/proto/kinfer.proto
+# Protobuf Python Version: 5.29.1
+"""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, 1, "", "kinfer/proto/kinfer.proto"
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\n\x19kinfer/proto/kinfer.proto\x12\x0ckinfer.proto"f\n\x12JointPositionValue\x12\x12\n\njoint_name\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x02\x12-\n\x04unit\x18\x03 \x01(\x0e\x32\x1f.kinfer.proto.JointPositionUnit"Z\n\x14JointPositionsSchema\x12-\n\x04unit\x18\x01 \x01(\x0e\x32\x1f.kinfer.proto.JointPositionUnit\x12\x13\n\x0bjoint_names\x18\x02 \x03(\t"G\n\x13JointPositionsValue\x12\x30\n\x06values\x18\x01 \x03(\x0b\x32 .kinfer.proto.JointPositionValue"f\n\x12JointVelocityValue\x12\x12\n\njoint_name\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x02\x12-\n\x04unit\x18\x03 \x01(\x0e\x32\x1f.kinfer.proto.JointVelocityUnit"[\n\x15JointVelocitiesSchema\x12-\n\x04unit\x18\x01 \x01(\x0e\x32\x1f.kinfer.proto.JointVelocityUnit\x12\x13\n\x0bjoint_names\x18\x02 \x03(\t"H\n\x14JointVelocitiesValue\x12\x30\n\x06values\x18\x01 \x03(\x0b\x32 .kinfer.proto.JointVelocityValue"b\n\x10JointTorqueValue\x12\x12\n\njoint_name\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x02\x12+\n\x04unit\x18\x03 \x01(\x0e\x32\x1d.kinfer.proto.JointTorqueUnit"V\n\x12JointTorquesSchema\x12+\n\x04unit\x18\x01 \x01(\x0e\x32\x1d.kinfer.proto.JointTorqueUnit\x12\x13\n\x0bjoint_names\x18\x02 \x03(\t"C\n\x11JointTorquesValue\x12.\n\x06values\x18\x01 \x03(\x0b\x32\x1e.kinfer.proto.JointTorqueValue"\xce\x01\n\x13JointCommandsSchema\x12\x13\n\x0bjoint_names\x18\x01 \x03(\t\x12\x32\n\x0btorque_unit\x18\x02 \x01(\x0e\x32\x1d.kinfer.proto.JointTorqueUnit\x12\x36\n\rvelocity_unit\x18\x03 \x01(\x0e\x32\x1f.kinfer.proto.JointVelocityUnit\x12\x36\n\rposition_unit\x18\x04 \x01(\x0e\x32\x1f.kinfer.proto.JointPositionUnit"\x97\x02\n\x11JointCommandValue\x12\x12\n\njoint_name\x18\x01 \x01(\t\x12\x0e\n\x06torque\x18\x02 \x01(\x02\x12\x10\n\x08velocity\x18\x03 \x01(\x02\x12\x10\n\x08position\x18\x04 \x01(\x02\x12\n\n\x02kp\x18\x05 \x01(\x02\x12\n\n\x02kd\x18\x06 \x01(\x02\x12\x32\n\x0btorque_unit\x18\x07 \x01(\x0e\x32\x1d.kinfer.proto.JointTorqueUnit\x12\x36\n\rvelocity_unit\x18\x08 \x01(\x0e\x32\x1f.kinfer.proto.JointVelocityUnit\x12\x36\n\rposition_unit\x18\t \x01(\x0e\x32\x1f.kinfer.proto.JointPositionUnit"E\n\x12JointCommandsValue\x12/\n\x06values\x18\x01 \x03(\x0b\x32\x1f.kinfer.proto.JointCommandValue"D\n\x11\x43\x61meraFrameSchema\x12\r\n\x05width\x18\x01 \x01(\x05\x12\x0e\n\x06height\x18\x02 \x01(\x05\x12\x10\n\x08\x63hannels\x18\x03 \x01(\x05" \n\x10\x43\x61meraFrameValue\x12\x0c\n\x04\x64\x61ta\x18\x01 \x01(\x0c"]\n\x10\x41udioFrameSchema\x12\x10\n\x08\x63hannels\x18\x01 \x01(\x05\x12\x13\n\x0bsample_rate\x18\x02 \x01(\x05\x12"\n\x05\x64type\x18\x03 \x01(\x0e\x32\x13.kinfer.proto.DType"\x1f\n\x0f\x41udioFrameValue\x12\x0c\n\x04\x64\x61ta\x18\x01 \x01(\x0c"8\n\x15ImuAccelerometerValue\x12\t\n\x01x\x18\x01 \x01(\x02\x12\t\n\x01y\x18\x02 \x01(\x02\x12\t\n\x01z\x18\x03 \x01(\x02"4\n\x11ImuGyroscopeValue\x12\t\n\x01x\x18\x01 \x01(\x02\x12\t\n\x01y\x18\x02 \x01(\x02\x12\t\n\x01z\x18\x03 \x01(\x02"7\n\x14ImuMagnetometerValue\x12\t\n\x01x\x18\x01 \x01(\x02\x12\t\n\x01y\x18\x02 \x01(\x02\x12\t\n\x01z\x18\x03 \x01(\x02"W\n\tImuSchema\x12\x19\n\x11use_accelerometer\x18\x01 \x01(\x08\x12\x15\n\ruse_gyroscope\x18\x02 \x01(\x08\x12\x18\n\x10use_magnetometer\x18\x03 \x01(\x08"\xc3\x01\n\x08ImuValue\x12@\n\x13linear_acceleration\x18\x01 \x01(\x0b\x32#.kinfer.proto.ImuAccelerometerValue\x12\x39\n\x10\x61ngular_velocity\x18\x02 \x01(\x0b\x32\x1f.kinfer.proto.ImuGyroscopeValue\x12:\n\x0emagnetic_field\x18\x03 \x01(\x0b\x32".kinfer.proto.ImuMagnetometerValue"=\n\x0fTimestampSchema\x12\x15\n\rstart_seconds\x18\x01 \x01(\x03\x12\x13\n\x0bstart_nanos\x18\x02 \x01(\x05"0\n\x0eTimestampValue\x12\x0f\n\x07seconds\x18\x01 \x01(\x03\x12\r\n\x05nanos\x18\x02 \x01(\x05")\n\x13VectorCommandSchema\x12\x12\n\ndimensions\x18\x01 \x01(\x05"$\n\x12VectorCommandValue\x12\x0e\n\x06values\x18\x01 \x03(\x02"F\n\x11StateTensorSchema\x12\r\n\x05shape\x18\x01 \x03(\x05\x12"\n\x05\x64type\x18\x02 \x01(\x0e\x32\x13.kinfer.proto.DType" \n\x10StateTensorValue\x12\x0c\n\x04\x64\x61ta\x18\x01 \x01(\x0c"\xd4\x04\n\x05Value\x12\x12\n\nvalue_name\x18\x01 \x01(\t\x12<\n\x0fjoint_positions\x18\x02 \x01(\x0b\x32!.kinfer.proto.JointPositionsValueH\x00\x12>\n\x10joint_velocities\x18\x03 \x01(\x0b\x32".kinfer.proto.JointVelocitiesValueH\x00\x12\x38\n\rjoint_torques\x18\x04 \x01(\x0b\x32\x1f.kinfer.proto.JointTorquesValueH\x00\x12:\n\x0ejoint_commands\x18\x05 \x01(\x0b\x32 .kinfer.proto.JointCommandsValueH\x00\x12\x36\n\x0c\x63\x61mera_frame\x18\x06 \x01(\x0b\x32\x1e.kinfer.proto.CameraFrameValueH\x00\x12\x34\n\x0b\x61udio_frame\x18\x07 \x01(\x0b\x32\x1d.kinfer.proto.AudioFrameValueH\x00\x12%\n\x03imu\x18\x08 \x01(\x0b\x32\x16.kinfer.proto.ImuValueH\x00\x12\x31\n\ttimestamp\x18\t \x01(\x0b\x32\x1c.kinfer.proto.TimestampValueH\x00\x12:\n\x0evector_command\x18\n \x01(\x0b\x32 .kinfer.proto.VectorCommandValueH\x00\x12\x36\n\x0cstate_tensor\x18\x0b \x01(\x0b\x32\x1e.kinfer.proto.StateTensorValueH\x00\x42\x07\n\x05value"\xe9\x04\n\x0bValueSchema\x12\x12\n\nvalue_name\x18\x01 \x01(\t\x12=\n\x0fjoint_positions\x18\x02 \x01(\x0b\x32".kinfer.proto.JointPositionsSchemaH\x00\x12?\n\x10joint_velocities\x18\x03 \x01(\x0b\x32#.kinfer.proto.JointVelocitiesSchemaH\x00\x12\x39\n\rjoint_torques\x18\x04 \x01(\x0b\x32 .kinfer.proto.JointTorquesSchemaH\x00\x12;\n\x0ejoint_commands\x18\x05 \x01(\x0b\x32!.kinfer.proto.JointCommandsSchemaH\x00\x12\x37\n\x0c\x63\x61mera_frame\x18\x06 \x01(\x0b\x32\x1f.kinfer.proto.CameraFrameSchemaH\x00\x12\x35\n\x0b\x61udio_frame\x18\x07 \x01(\x0b\x32\x1e.kinfer.proto.AudioFrameSchemaH\x00\x12&\n\x03imu\x18\x08 \x01(\x0b\x32\x17.kinfer.proto.ImuSchemaH\x00\x12\x32\n\ttimestamp\x18\t \x01(\x0b\x32\x1d.kinfer.proto.TimestampSchemaH\x00\x12;\n\x0evector_command\x18\n \x01(\x0b\x32!.kinfer.proto.VectorCommandSchemaH\x00\x12\x37\n\x0cstate_tensor\x18\x0b \x01(\x0b\x32\x1f.kinfer.proto.StateTensorSchemaH\x00\x42\x0c\n\nvalue_type"5\n\x08IOSchema\x12)\n\x06values\x18\x01 \x03(\x0b\x32\x19.kinfer.proto.ValueSchema")\n\x02IO\x12#\n\x06values\x18\x01 \x03(\x0b\x32\x13.kinfer.proto.Value"j\n\x0bModelSchema\x12,\n\x0cinput_schema\x18\x01 \x01(\x0b\x32\x16.kinfer.proto.IOSchema\x12-\n\routput_schema\x18\x02 \x01(\x0b\x32\x16.kinfer.proto.IOSchema*\x88\x01\n\x05\x44Type\x12\x07\n\x03\x46P8\x10\x00\x12\x08\n\x04\x46P16\x10\x01\x12\x08\n\x04\x46P32\x10\x02\x12\x08\n\x04\x46P64\x10\x03\x12\x08\n\x04INT8\x10\x04\x12\t\n\x05INT16\x10\x05\x12\t\n\x05INT32\x10\x06\x12\t\n\x05INT64\x10\x07\x12\t\n\x05UINT8\x10\x08\x12\n\n\x06UINT16\x10\t\x12\n\n\x06UINT32\x10\n\x12\n\n\x06UINT64\x10\x0b*-\n\x11JointPositionUnit\x12\x0b\n\x07\x44\x45GREES\x10\x00\x12\x0b\n\x07RADIANS\x10\x01*C\n\x11JointVelocityUnit\x12\x16\n\x12\x44\x45GREES_PER_SECOND\x10\x00\x12\x16\n\x12RADIANS_PER_SECOND\x10\x01*$\n\x0fJointTorqueUnit\x12\x11\n\rNEWTON_METERS\x10\x00\x62\x06proto3'
+)
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "kinfer.proto.kinfer_pb2", _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+    DESCRIPTOR._loaded_options = None
+    _globals["_DTYPE"]._serialized_start = 3816
+    _globals["_DTYPE"]._serialized_end = 3952
+    _globals["_JOINTPOSITIONUNIT"]._serialized_start = 3954
+    _globals["_JOINTPOSITIONUNIT"]._serialized_end = 3999
+    _globals["_JOINTVELOCITYUNIT"]._serialized_start = 4001
+    _globals["_JOINTVELOCITYUNIT"]._serialized_end = 4068
+    _globals["_JOINTTORQUEUNIT"]._serialized_start = 4070
+    _globals["_JOINTTORQUEUNIT"]._serialized_end = 4106
+    _globals["_JOINTPOSITIONVALUE"]._serialized_start = 43
+    _globals["_JOINTPOSITIONVALUE"]._serialized_end = 145
+    _globals["_JOINTPOSITIONSSCHEMA"]._serialized_start = 147
+    _globals["_JOINTPOSITIONSSCHEMA"]._serialized_end = 237
+    _globals["_JOINTPOSITIONSVALUE"]._serialized_start = 239
+    _globals["_JOINTPOSITIONSVALUE"]._serialized_end = 310
+    _globals["_JOINTVELOCITYVALUE"]._serialized_start = 312
+    _globals["_JOINTVELOCITYVALUE"]._serialized_end = 414
+    _globals["_JOINTVELOCITIESSCHEMA"]._serialized_start = 416
+    _globals["_JOINTVELOCITIESSCHEMA"]._serialized_end = 507
+    _globals["_JOINTVELOCITIESVALUE"]._serialized_start = 509
+    _globals["_JOINTVELOCITIESVALUE"]._serialized_end = 581
+    _globals["_JOINTTORQUEVALUE"]._serialized_start = 583
+    _globals["_JOINTTORQUEVALUE"]._serialized_end = 681
+    _globals["_JOINTTORQUESSCHEMA"]._serialized_start = 683
+    _globals["_JOINTTORQUESSCHEMA"]._serialized_end = 769
+    _globals["_JOINTTORQUESVALUE"]._serialized_start = 771
+    _globals["_JOINTTORQUESVALUE"]._serialized_end = 838
+    _globals["_JOINTCOMMANDSSCHEMA"]._serialized_start = 841
+    _globals["_JOINTCOMMANDSSCHEMA"]._serialized_end = 1047
+    _globals["_JOINTCOMMANDVALUE"]._serialized_start = 1050
+    _globals["_JOINTCOMMANDVALUE"]._serialized_end = 1329
+    _globals["_JOINTCOMMANDSVALUE"]._serialized_start = 1331
+    _globals["_JOINTCOMMANDSVALUE"]._serialized_end = 1400
+    _globals["_CAMERAFRAMESCHEMA"]._serialized_start = 1402
+    _globals["_CAMERAFRAMESCHEMA"]._serialized_end = 1470
+    _globals["_CAMERAFRAMEVALUE"]._serialized_start = 1472
+    _globals["_CAMERAFRAMEVALUE"]._serialized_end = 1504
+    _globals["_AUDIOFRAMESCHEMA"]._serialized_start = 1506
+    _globals["_AUDIOFRAMESCHEMA"]._serialized_end = 1599
+    _globals["_AUDIOFRAMEVALUE"]._serialized_start = 1601
+    _globals["_AUDIOFRAMEVALUE"]._serialized_end = 1632
+    _globals["_IMUACCELEROMETERVALUE"]._serialized_start = 1634
+    _globals["_IMUACCELEROMETERVALUE"]._serialized_end = 1690
+    _globals["_IMUGYROSCOPEVALUE"]._serialized_start = 1692
+    _globals["_IMUGYROSCOPEVALUE"]._serialized_end = 1744
+    _globals["_IMUMAGNETOMETERVALUE"]._serialized_start = 1746
+    _globals["_IMUMAGNETOMETERVALUE"]._serialized_end = 1801
+    _globals["_IMUSCHEMA"]._serialized_start = 1803
+    _globals["_IMUSCHEMA"]._serialized_end = 1890
+    _globals["_IMUVALUE"]._serialized_start = 1893
+    _globals["_IMUVALUE"]._serialized_end = 2088
+    _globals["_TIMESTAMPSCHEMA"]._serialized_start = 2090
+    _globals["_TIMESTAMPSCHEMA"]._serialized_end = 2151
+    _globals["_TIMESTAMPVALUE"]._serialized_start = 2153
+    _globals["_TIMESTAMPVALUE"]._serialized_end = 2201
+    _globals["_VECTORCOMMANDSCHEMA"]._serialized_start = 2203
+    _globals["_VECTORCOMMANDSCHEMA"]._serialized_end = 2244
+    _globals["_VECTORCOMMANDVALUE"]._serialized_start = 2246
+    _globals["_VECTORCOMMANDVALUE"]._serialized_end = 2282
+    _globals["_STATETENSORSCHEMA"]._serialized_start = 2284
+    _globals["_STATETENSORSCHEMA"]._serialized_end = 2354
+    _globals["_STATETENSORVALUE"]._serialized_start = 2356
+    _globals["_STATETENSORVALUE"]._serialized_end = 2388
+    _globals["_VALUE"]._serialized_start = 2391
+    _globals["_VALUE"]._serialized_end = 2987
+    _globals["_VALUESCHEMA"]._serialized_start = 2990
+    _globals["_VALUESCHEMA"]._serialized_end = 3607
+    _globals["_IOSCHEMA"]._serialized_start = 3609
+    _globals["_IOSCHEMA"]._serialized_end = 3662
+    _globals["_IO"]._serialized_start = 3664
+    _globals["_IO"]._serialized_end = 3705
+    _globals["_MODELSCHEMA"]._serialized_start = 3707
+    _globals["_MODELSCHEMA"]._serialized_end = 3813
+# @@protoc_insertion_point(module_scope)