Source code for viam.gen.app.mltraining.v1.ml_training_grpc

import abc
import typing
import grpclib.const
import grpclib.client
import grpclib.exceptions
if typing.TYPE_CHECKING:
    import grpclib.server
import google.protobuf.timestamp_pb2
import google.rpc.status_pb2
from .... import tagger
from .... import app


[docs]
class MLTrainingServiceBase(abc.ABC):


[docs]
    @abc.abstractmethod
    async def SubmitTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.SubmitTrainingJobRequest, app.mltraining.v1.ml_training_pb2.SubmitTrainingJobResponse]') -> None:
        pass



[docs]
    @abc.abstractmethod
    async def SubmitCustomTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.SubmitCustomTrainingJobRequest, app.mltraining.v1.ml_training_pb2.SubmitCustomTrainingJobResponse]') -> None:
        pass



[docs]
    @abc.abstractmethod
    async def GetTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.GetTrainingJobRequest, app.mltraining.v1.ml_training_pb2.GetTrainingJobResponse]') -> None:
        pass



[docs]
    @abc.abstractmethod
    async def ListTrainingJobs(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.ListTrainingJobsRequest, app.mltraining.v1.ml_training_pb2.ListTrainingJobsResponse]') -> None:
        pass



[docs]
    @abc.abstractmethod
    async def CancelTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.CancelTrainingJobRequest, app.mltraining.v1.ml_training_pb2.CancelTrainingJobResponse]') -> None:
        pass



[docs]
    @abc.abstractmethod
    async def DeleteCompletedTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.DeleteCompletedTrainingJobRequest, app.mltraining.v1.ml_training_pb2.DeleteCompletedTrainingJobResponse]') -> None:
        pass



[docs]
    @abc.abstractmethod
    async def GetTrainingJobLogs(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsRequest, app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsResponse]') -> None:
        pass



[docs]
    @abc.abstractmethod
    async def ListSupportedContainers(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.ListSupportedContainersRequest, app.mltraining.v1.ml_training_pb2.ListSupportedContainersResponse]') -> None:
        pass



[docs]
    @abc.abstractmethod
    async def ListContainers(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.ListContainersRequest, app.mltraining.v1.ml_training_pb2.ListContainersResponse]') -> None:
        pass



[docs]
    @abc.abstractmethod
    async def GetContainer(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.GetContainerRequest, app.mltraining.v1.ml_training_pb2.GetContainerResponse]') -> None:
        pass



[docs]
    @abc.abstractmethod
    async def RegisterCustomTrainingContainer(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.RegisterCustomTrainingContainerRequest, app.mltraining.v1.ml_training_pb2.RegisterCustomTrainingContainerResponse]') -> None:
        pass



[docs]
    @abc.abstractmethod
    async def DeleteCustomTrainingContainer(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.DeleteCustomTrainingContainerRequest, app.mltraining.v1.ml_training_pb2.DeleteCustomTrainingContainerResponse]') -> None:
        pass



[docs]
    def __mapping__(self) -> typing.Dict[str, grpclib.const.Handler]:
        return {'/viam.app.mltraining.v1.MLTrainingService/SubmitTrainingJob': grpclib.const.Handler(self.SubmitTrainingJob, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.SubmitTrainingJobRequest, app.mltraining.v1.ml_training_pb2.SubmitTrainingJobResponse), '/viam.app.mltraining.v1.MLTrainingService/SubmitCustomTrainingJob': grpclib.const.Handler(self.SubmitCustomTrainingJob, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.SubmitCustomTrainingJobRequest, app.mltraining.v1.ml_training_pb2.SubmitCustomTrainingJobResponse), '/viam.app.mltraining.v1.MLTrainingService/GetTrainingJob': grpclib.const.Handler(self.GetTrainingJob, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.GetTrainingJobRequest, app.mltraining.v1.ml_training_pb2.GetTrainingJobResponse), '/viam.app.mltraining.v1.MLTrainingService/ListTrainingJobs': grpclib.const.Handler(self.ListTrainingJobs, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.ListTrainingJobsRequest, app.mltraining.v1.ml_training_pb2.ListTrainingJobsResponse), '/viam.app.mltraining.v1.MLTrainingService/CancelTrainingJob': grpclib.const.Handler(self.CancelTrainingJob, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.CancelTrainingJobRequest, app.mltraining.v1.ml_training_pb2.CancelTrainingJobResponse), '/viam.app.mltraining.v1.MLTrainingService/DeleteCompletedTrainingJob': grpclib.const.Handler(self.DeleteCompletedTrainingJob, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.DeleteCompletedTrainingJobRequest, app.mltraining.v1.ml_training_pb2.DeleteCompletedTrainingJobResponse), '/viam.app.mltraining.v1.MLTrainingService/GetTrainingJobLogs': grpclib.const.Handler(self.GetTrainingJobLogs, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsRequest, app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsResponse), '/viam.app.mltraining.v1.MLTrainingService/ListSupportedContainers': grpclib.const.Handler(self.ListSupportedContainers, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.ListSupportedContainersRequest, app.mltraining.v1.ml_training_pb2.ListSupportedContainersResponse), '/viam.app.mltraining.v1.MLTrainingService/ListContainers': grpclib.const.Handler(self.ListContainers, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.ListContainersRequest, app.mltraining.v1.ml_training_pb2.ListContainersResponse), '/viam.app.mltraining.v1.MLTrainingService/GetContainer': grpclib.const.Handler(self.GetContainer, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.GetContainerRequest, app.mltraining.v1.ml_training_pb2.GetContainerResponse), '/viam.app.mltraining.v1.MLTrainingService/RegisterCustomTrainingContainer': grpclib.const.Handler(self.RegisterCustomTrainingContainer, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.RegisterCustomTrainingContainerRequest, app.mltraining.v1.ml_training_pb2.RegisterCustomTrainingContainerResponse), '/viam.app.mltraining.v1.MLTrainingService/DeleteCustomTrainingContainer': grpclib.const.Handler(self.DeleteCustomTrainingContainer, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.DeleteCustomTrainingContainerRequest, app.mltraining.v1.ml_training_pb2.DeleteCustomTrainingContainerResponse)}




[docs]
class UnimplementedMLTrainingServiceBase(MLTrainingServiceBase):


[docs]
    async def SubmitTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.SubmitTrainingJobRequest, app.mltraining.v1.ml_training_pb2.SubmitTrainingJobResponse]') -> None:
        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)



[docs]
    async def SubmitCustomTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.SubmitCustomTrainingJobRequest, app.mltraining.v1.ml_training_pb2.SubmitCustomTrainingJobResponse]') -> None:
        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)



[docs]
    async def GetTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.GetTrainingJobRequest, app.mltraining.v1.ml_training_pb2.GetTrainingJobResponse]') -> None:
        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)



[docs]
    async def ListTrainingJobs(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.ListTrainingJobsRequest, app.mltraining.v1.ml_training_pb2.ListTrainingJobsResponse]') -> None:
        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)



[docs]
    async def CancelTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.CancelTrainingJobRequest, app.mltraining.v1.ml_training_pb2.CancelTrainingJobResponse]') -> None:
        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)



[docs]
    async def DeleteCompletedTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.DeleteCompletedTrainingJobRequest, app.mltraining.v1.ml_training_pb2.DeleteCompletedTrainingJobResponse]') -> None:
        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)



[docs]
    async def GetTrainingJobLogs(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsRequest, app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsResponse]') -> None:
        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)



[docs]
    async def ListSupportedContainers(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.ListSupportedContainersRequest, app.mltraining.v1.ml_training_pb2.ListSupportedContainersResponse]') -> None:
        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)



[docs]
    async def ListContainers(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.ListContainersRequest, app.mltraining.v1.ml_training_pb2.ListContainersResponse]') -> None:
        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)



[docs]
    async def GetContainer(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.GetContainerRequest, app.mltraining.v1.ml_training_pb2.GetContainerResponse]') -> None:
        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)



[docs]
    async def RegisterCustomTrainingContainer(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.RegisterCustomTrainingContainerRequest, app.mltraining.v1.ml_training_pb2.RegisterCustomTrainingContainerResponse]') -> None:
        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)



[docs]
    async def DeleteCustomTrainingContainer(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.DeleteCustomTrainingContainerRequest, app.mltraining.v1.ml_training_pb2.DeleteCustomTrainingContainerResponse]') -> None:
        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)




[docs]
class MLTrainingServiceStub:

    def __init__(self, channel: grpclib.client.Channel) -> None:
        self.SubmitTrainingJob = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/SubmitTrainingJob', app.mltraining.v1.ml_training_pb2.SubmitTrainingJobRequest, app.mltraining.v1.ml_training_pb2.SubmitTrainingJobResponse)
        self.SubmitCustomTrainingJob = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/SubmitCustomTrainingJob', app.mltraining.v1.ml_training_pb2.SubmitCustomTrainingJobRequest, app.mltraining.v1.ml_training_pb2.SubmitCustomTrainingJobResponse)
        self.GetTrainingJob = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/GetTrainingJob', app.mltraining.v1.ml_training_pb2.GetTrainingJobRequest, app.mltraining.v1.ml_training_pb2.GetTrainingJobResponse)
        self.ListTrainingJobs = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/ListTrainingJobs', app.mltraining.v1.ml_training_pb2.ListTrainingJobsRequest, app.mltraining.v1.ml_training_pb2.ListTrainingJobsResponse)
        self.CancelTrainingJob = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/CancelTrainingJob', app.mltraining.v1.ml_training_pb2.CancelTrainingJobRequest, app.mltraining.v1.ml_training_pb2.CancelTrainingJobResponse)
        self.DeleteCompletedTrainingJob = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/DeleteCompletedTrainingJob', app.mltraining.v1.ml_training_pb2.DeleteCompletedTrainingJobRequest, app.mltraining.v1.ml_training_pb2.DeleteCompletedTrainingJobResponse)
        self.GetTrainingJobLogs = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/GetTrainingJobLogs', app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsRequest, app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsResponse)
        self.ListSupportedContainers = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/ListSupportedContainers', app.mltraining.v1.ml_training_pb2.ListSupportedContainersRequest, app.mltraining.v1.ml_training_pb2.ListSupportedContainersResponse)
        self.ListContainers = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/ListContainers', app.mltraining.v1.ml_training_pb2.ListContainersRequest, app.mltraining.v1.ml_training_pb2.ListContainersResponse)
        self.GetContainer = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/GetContainer', app.mltraining.v1.ml_training_pb2.GetContainerRequest, app.mltraining.v1.ml_training_pb2.GetContainerResponse)
        self.RegisterCustomTrainingContainer = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/RegisterCustomTrainingContainer', app.mltraining.v1.ml_training_pb2.RegisterCustomTrainingContainerRequest, app.mltraining.v1.ml_training_pb2.RegisterCustomTrainingContainerResponse)
        self.DeleteCustomTrainingContainer = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/DeleteCustomTrainingContainer', app.mltraining.v1.ml_training_pb2.DeleteCustomTrainingContainerRequest, app.mltraining.v1.ml_training_pb2.DeleteCustomTrainingContainerResponse)