1- from typing import Dict , Mapping , Optional
1+ from typing import Dict , Optional
22
33from grpclib .client import Channel
44from numpy .typing import NDArray
55
6- from viam .proto .common import DoCommandRequest , DoCommandResponse
76from viam .proto .service .mlmodel import InferRequest , InferResponse , MetadataRequest , MetadataResponse , MLModelServiceStub
87from viam .resource .rpc_client_base import ReconfigurableResourceRPCClientBase
98from viam .services .mlmodel .utils import flat_tensors_to_ndarrays , ndarrays_to_flat_tensors
10- from viam .utils import ValueTypes , dict_to_struct , struct_to_dict
119
1210from .mlmodel import Metadata , MLModel
1311
@@ -27,8 +25,3 @@ async def metadata(self, *, timeout: Optional[float] = None) -> Metadata:
2725 request = MetadataRequest (name = self .name )
2826 response : MetadataResponse = await self .client .Metadata (request )
2927 return response .metadata
30-
31- async def do_command (self , command : Mapping [str , ValueTypes ], * , timeout : Optional [float ] = None , ** kwargs ) -> Mapping [str , ValueTypes ]:
32- request = DoCommandRequest (name = self .name , command = dict_to_struct (command ))
33- response : DoCommandResponse = await self .client .DoCommand (request , timeout = timeout )
34- return struct_to_dict (response .result )
0 commit comments