Module Contents



MLModel represents a Machine Learning Model service.

class str, channel: grpclib.client.Channel)[source]

Bases:, viam.resource.rpc_client_base.ReconfigurableResourceRPCClientBase

MLModel represents a Machine Learning Model service.

This acts as an abstract base class for any drivers representing specific arm implementations. This cannot be used on its own. If the __init__() function is overridden, it must call the super().__init__() function.

async infer(input_tensors: Dict[str, numpy.typing.NDArray], *, timeout: Optional[float] = None) Dict[str, numpy.typing.NDArray][source]

Take an already ordered input tensor as an array, make an inference on the model, and return an output tensor map.


input_tensors (Dict[str, NDArray]) – A dictionary of input flat tensors as specified in the metadata


A dictionary of output flat tensors as specified in the metadata

Return type:

Dict[str, NDArray]

async metadata(*, timeout: Optional[float] = None)[source]

Get the metadata (such as name, type, expected tensor/array shape, inputs, and outputs) associated with the ML model.


The metadata

Return type:


async do_command(command: Mapping[str, viam.utils.ValueTypes], *, timeout: Optional[float] = None, **kwargs) Mapping[str, viam.utils.ValueTypes][source]

Send/Receive arbitrary commands to the Resource


command (Mapping[str, ValueTypes]) – The command to execute


NotImplementedError – Raised if the Resource does not support arbitrary commands


Result of the executed command

Return type:

Mapping[str, ValueTypes]

classmethod from_robot(robot: viam.robot.client.RobotClient, name: str) typing_extensions.Self

Get the service named name from the provided robot.

  • robot (RobotClient) – The robot

  • name (str) – The name of the service


The service, if it exists on the robot

Return type:


classmethod get_resource_name(name: str) viam.proto.common.ResourceName

Get the ResourceName for this Resource with the given name


name (str) – The name of the Resource

get_operation(kwargs: Mapping[str, Any]) viam.operations.Operation

Get the Operation associated with the currently running function.

When writing custom resources, you should get the Operation by calling this function and check to see if it’s cancelled. If the Operation is cancelled, then you can perform any necessary (terminating long running tasks, cleaning up connections, etc. ).


kwargs (Mapping[str, Any]) – The kwargs object containing the operation


The operation associated with this function

Return type:


async close()

Safely shut down the resource and prevent further use.

Close must be idempotent. Later configuration may allow a resource to be “open” again. If a resource does not want or need a close function, it is assumed that the resource does not need to retun errors when future non-Close methods are called.