viam-sdk 0.3.0__py3-none-linux_armv6l.whl → 0.66.0__py3-none-linux_armv6l.whl

viam/gen/app/mlinference/v1/ml_inference_pb2.pyi

@@ -0,0 +1,63 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+from .... import app
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+from .... import service
+import typing
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class GetInferenceRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    REGISTRY_ITEM_ID_FIELD_NUMBER: builtins.int
+    REGISTRY_ITEM_VERSION_FIELD_NUMBER: builtins.int
+    BINARY_ID_FIELD_NUMBER: builtins.int
+    BINARY_DATA_ID_FIELD_NUMBER: builtins.int
+    ORGANIZATION_ID_FIELD_NUMBER: builtins.int
+    registry_item_id: builtins.str
+    'The model framework and model type are inferred from the ML model registry item;\n    For valid model types (classification, detections) we will return the formatted\n    labels or annotations from the associated inference outputs.\n    '
+    registry_item_version: builtins.str
+    binary_data_id: builtins.str
+    organization_id: builtins.str
+
+    @property
+    def binary_id(self) -> app.data.v1.data_pb2.BinaryID:
+        ...
+
+    def __init__(self, *, registry_item_id: builtins.str=..., registry_item_version: builtins.str=..., binary_id: app.data.v1.data_pb2.BinaryID | None=..., binary_data_id: builtins.str=..., organization_id: builtins.str=...) -> None:
+        ...
+
+    def HasField(self, field_name: typing.Literal['binary_id', b'binary_id']) -> builtins.bool:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['binary_data_id', b'binary_data_id', 'binary_id', b'binary_id', 'organization_id', b'organization_id', 'registry_item_id', b'registry_item_id', 'registry_item_version', b'registry_item_version']) -> None:
+        ...
+global___GetInferenceRequest = GetInferenceRequest
+
+@typing.final
+class GetInferenceResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    OUTPUT_TENSORS_FIELD_NUMBER: builtins.int
+    ANNOTATIONS_FIELD_NUMBER: builtins.int
+
+    @property
+    def output_tensors(self) -> service.mlmodel.v1.mlmodel_pb2.FlatTensors:
+        ...
+
+    @property
+    def annotations(self) -> app.data.v1.data_pb2.Annotations:
+        ...
+
+    def __init__(self, *, output_tensors: service.mlmodel.v1.mlmodel_pb2.FlatTensors | None=..., annotations: app.data.v1.data_pb2.Annotations | None=...) -> None:
+        ...
+
+    def HasField(self, field_name: typing.Literal['annotations', b'annotations', 'output_tensors', b'output_tensors']) -> builtins.bool:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['annotations', b'annotations', 'output_tensors', b'output_tensors']) -> None:
+        ...
+global___GetInferenceResponse = GetInferenceResponse

viam/gen/app/mltraining/v1/ml_training_grpc.py

@@ -2,11 +2,13 @@ import abc
 import typing
 import grpclib.const
 import grpclib.client
+import grpclib.exceptions
 if typing.TYPE_CHECKING:
     import grpclib.server
-from .... import app
 import google.protobuf.timestamp_pb2
+import google.rpc.status_pb2
 from .... import tagger
+from .... import app
 
 class MLTrainingServiceBase(abc.ABC):
 
@@ -14,6 +16,10 @@ class MLTrainingServiceBase(abc.ABC):
     async def SubmitTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.SubmitTrainingJobRequest, app.mltraining.v1.ml_training_pb2.SubmitTrainingJobResponse]') -> None:
         pass
 
+    @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
+
     @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
@@ -26,13 +32,55 @@ class MLTrainingServiceBase(abc.ABC):
     async def CancelTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.CancelTrainingJobRequest, app.mltraining.v1.ml_training_pb2.CancelTrainingJobResponse]') -> None:
         pass
 
+    @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
+
+    @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
+
+    @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
+
     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/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)}
+        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)}
+
+class UnimplementedMLTrainingServiceBase(MLTrainingServiceBase):
+
+    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)
+
+    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)
+
+    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)
+
+    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)
+
+    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)
+
+    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)
+
+    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)
+
+    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)
 
 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.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)

viam/gen/app/mltraining/v1/ml_training_pb2.py

@@ -1,65 +1,142 @@
 """Generated protocol buffer code."""
-from google.protobuf.internal import builder as _builder
 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, 2, '', 'app/mltraining/v1/ml_training.proto')
 _sym_db = _symbol_database.Default()
-from ....app.data.v1 import data_pb2 as app_dot_data_dot_v1_dot_data__pb2
 from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+from google.rpc import status_pb2 as google_dot_rpc_dot_status__pb2
 from ....tagger.v1 import tagger_pb2 as tagger_dot_v1_dot_tagger__pb2
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n#app/mltraining/v1/ml_training.proto\x12\x16viam.app.mltraining.v1\x1a\x16app/data/v1/data.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x16tagger/v1/tagger.proto"\x87\x04\n\x18SubmitTrainingJobRequest\x12R\n\x06filter\x18\x01 \x01(\x0b2\x18.viam.app.data.v1.FilterB \x9a\x84\x9e\x03\x1bbson:"filter" json:"filter"R\x06filter\x12[\n\x0forganization_id\x18\x02 \x01(\tB2\x9a\x84\x9e\x03-bson:"organization_id" json:"organization_id"R\x0eorganizationId\x12G\n\nmodel_name\x18\x03 \x01(\tB(\x9a\x84\x9e\x03#bson:"model_name" json:"model_name"R\tmodelName\x12S\n\rmodel_version\x18\x04 \x01(\tB.\x9a\x84\x9e\x03)bson:"model_version" json:"model_version"R\x0cmodelVersion\x12j\n\nmodel_type\x18\x05 \x01(\x0e2!.viam.app.mltraining.v1.ModelTypeB(\x9a\x84\x9e\x03#bson:"model_type" json:"model_type"R\tmodelType\x120\n\x04tags\x18\x06 \x03(\tB\x1c\x9a\x84\x9e\x03\x17bson:"tags" json:"tags"R\x04tags"+\n\x19SubmitTrainingJobResponse\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"\'\n\x15GetTrainingJobRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"a\n\x16GetTrainingJobResponse\x12G\n\x08metadata\x18\x01 \x01(\x0b2+.viam.app.mltraining.v1.TrainingJobMetadataR\x08metadata"\x82\x01\n\x17ListTrainingJobsRequest\x12\'\n\x0forganization_id\x18\x01 \x01(\tR\x0eorganizationId\x12>\n\x06status\x18\x02 \x01(\x0e2&.viam.app.mltraining.v1.TrainingStatusR\x06status"[\n\x18ListTrainingJobsResponse\x12?\n\x04jobs\x18\x01 \x03(\x0b2+.viam.app.mltraining.v1.TrainingJobMetadataR\x04jobs"\x97\x05\n\x13TrainingJobMetadata\x12n\n\x07request\x18\x01 \x01(\x0b20.viam.app.mltraining.v1.SubmitTrainingJobRequestB"\x9a\x84\x9e\x03\x1dbson:"request" json:"request"R\x07request\x12`\n\x06status\x18\x02 \x01(\x0e2&.viam.app.mltraining.v1.TrainingStatusB \x9a\x84\x9e\x03\x1bbson:"status" json:"status"R\x06status\x12c\n\ncreated_on\x18\x03 \x01(\x0b2\x1a.google.protobuf.TimestampB(\x9a\x84\x9e\x03#bson:"created_on" json:"created_on"R\tcreatedOn\x12o\n\rlast_modified\x18\x04 \x01(\x0b2\x1a.google.protobuf.TimestampB.\x9a\x84\x9e\x03)bson:"last_modified" json:"last_modified"R\x0clastModified\x12Z\n\x0fsynced_model_id\x18\x05 \x01(\tB2\x9a\x84\x9e\x03-bson:"synced_model_id" json:"synced_model_id"R\rsyncedModelId\x12G\n\nuser_email\x18\x06 \x01(\tB(\x9a\x84\x9e\x03#bson:"user_email" json:"user_email"R\tuserEmail\x123\n\x02id\x18\x07 \x01(\tB#\x9a\x84\x9e\x03\x1ebson:"_id" json:"id,omitempty"R\x02id"*\n\x18CancelTrainingJobRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"\x1b\n\x19CancelTrainingJobResponse*\x9f\x01\n\tModelType\x12\x1a\n\x16MODEL_TYPE_UNSPECIFIED\x10\x00\x12*\n&MODEL_TYPE_SINGLE_LABEL_CLASSIFICATION\x10\x01\x12)\n%MODEL_TYPE_MULTI_LABEL_CLASSIFICATION\x10\x02\x12\x1f\n\x1bMODEL_TYPE_OBJECT_DETECTION\x10\x03*\xe7\x01\n\x0eTrainingStatus\x12\x1f\n\x1bTRAINING_STATUS_UNSPECIFIED\x10\x00\x12\x1b\n\x17TRAINING_STATUS_PENDING\x10\x01\x12\x1f\n\x1bTRAINING_STATUS_IN_PROGRESS\x10\x02\x12\x1d\n\x19TRAINING_STATUS_COMPLETED\x10\x03\x12\x1a\n\x16TRAINING_STATUS_FAILED\x10\x04\x12\x1c\n\x18TRAINING_STATUS_CANCELED\x10\x05\x12\x1d\n\x19TRAINING_STATUS_CANCELING\x10\x062\xef\x03\n\x11MLTrainingService\x12x\n\x11SubmitTrainingJob\x120.viam.app.mltraining.v1.SubmitTrainingJobRequest\x1a1.viam.app.mltraining.v1.SubmitTrainingJobResponse\x12o\n\x0eGetTrainingJob\x12-.viam.app.mltraining.v1.GetTrainingJobRequest\x1a..viam.app.mltraining.v1.GetTrainingJobResponse\x12u\n\x10ListTrainingJobs\x12/.viam.app.mltraining.v1.ListTrainingJobsRequest\x1a0.viam.app.mltraining.v1.ListTrainingJobsResponse\x12x\n\x11CancelTrainingJob\x120.viam.app.mltraining.v1.CancelTrainingJobRequest\x1a1.viam.app.mltraining.v1.CancelTrainingJobResponseB#Z!go.viam.com/api/app/mltraining/v1b\x06proto3')
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'app.mltraining.v1.ml_training_pb2', globals())
-if _descriptor._USE_C_DESCRIPTORS == False:
-    DESCRIPTOR._options = None
-    DESCRIPTOR._serialized_options = b'Z!go.viam.com/api/app/mltraining/v1'
-    _SUBMITTRAININGJOBREQUEST.fields_by_name['filter']._options = None
-    _SUBMITTRAININGJOBREQUEST.fields_by_name['filter']._serialized_options = b'\x9a\x84\x9e\x03\x1bbson:"filter" json:"filter"'
-    _SUBMITTRAININGJOBREQUEST.fields_by_name['organization_id']._options = None
-    _SUBMITTRAININGJOBREQUEST.fields_by_name['organization_id']._serialized_options = b'\x9a\x84\x9e\x03-bson:"organization_id" json:"organization_id"'
-    _SUBMITTRAININGJOBREQUEST.fields_by_name['model_name']._options = None
-    _SUBMITTRAININGJOBREQUEST.fields_by_name['model_name']._serialized_options = b'\x9a\x84\x9e\x03#bson:"model_name" json:"model_name"'
-    _SUBMITTRAININGJOBREQUEST.fields_by_name['model_version']._options = None
-    _SUBMITTRAININGJOBREQUEST.fields_by_name['model_version']._serialized_options = b'\x9a\x84\x9e\x03)bson:"model_version" json:"model_version"'
-    _SUBMITTRAININGJOBREQUEST.fields_by_name['model_type']._options = None
-    _SUBMITTRAININGJOBREQUEST.fields_by_name['model_type']._serialized_options = b'\x9a\x84\x9e\x03#bson:"model_type" json:"model_type"'
-    _SUBMITTRAININGJOBREQUEST.fields_by_name['tags']._options = None
-    _SUBMITTRAININGJOBREQUEST.fields_by_name['tags']._serialized_options = b'\x9a\x84\x9e\x03\x17bson:"tags" json:"tags"'
-    _TRAININGJOBMETADATA.fields_by_name['request']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['request']._serialized_options = b'\x9a\x84\x9e\x03\x1dbson:"request" json:"request"'
-    _TRAININGJOBMETADATA.fields_by_name['status']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['status']._serialized_options = b'\x9a\x84\x9e\x03\x1bbson:"status" json:"status"'
-    _TRAININGJOBMETADATA.fields_by_name['created_on']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['created_on']._serialized_options = b'\x9a\x84\x9e\x03#bson:"created_on" json:"created_on"'
-    _TRAININGJOBMETADATA.fields_by_name['last_modified']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['last_modified']._serialized_options = b'\x9a\x84\x9e\x03)bson:"last_modified" json:"last_modified"'
-    _TRAININGJOBMETADATA.fields_by_name['synced_model_id']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['synced_model_id']._serialized_options = b'\x9a\x84\x9e\x03-bson:"synced_model_id" json:"synced_model_id"'
-    _TRAININGJOBMETADATA.fields_by_name['user_email']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['user_email']._serialized_options = b'\x9a\x84\x9e\x03#bson:"user_email" json:"user_email"'
-    _TRAININGJOBMETADATA.fields_by_name['id']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['id']._serialized_options = b'\x9a\x84\x9e\x03\x1ebson:"_id" json:"id,omitempty"'
-    _MODELTYPE._serialized_start = 1817
-    _MODELTYPE._serialized_end = 1976
-    _TRAININGSTATUS._serialized_start = 1979
-    _TRAININGSTATUS._serialized_end = 2210
-    _SUBMITTRAININGJOBREQUEST._serialized_start = 145
-    _SUBMITTRAININGJOBREQUEST._serialized_end = 664
-    _SUBMITTRAININGJOBRESPONSE._serialized_start = 666
-    _SUBMITTRAININGJOBRESPONSE._serialized_end = 709
-    _GETTRAININGJOBREQUEST._serialized_start = 711
-    _GETTRAININGJOBREQUEST._serialized_end = 750
-    _GETTRAININGJOBRESPONSE._serialized_start = 752
-    _GETTRAININGJOBRESPONSE._serialized_end = 849
-    _LISTTRAININGJOBSREQUEST._serialized_start = 852
-    _LISTTRAININGJOBSREQUEST._serialized_end = 982
-    _LISTTRAININGJOBSRESPONSE._serialized_start = 984
-    _LISTTRAININGJOBSRESPONSE._serialized_end = 1075
-    _TRAININGJOBMETADATA._serialized_start = 1078
-    _TRAININGJOBMETADATA._serialized_end = 1741
-    _CANCELTRAININGJOBREQUEST._serialized_start = 1743
-    _CANCELTRAININGJOBREQUEST._serialized_end = 1785
-    _CANCELTRAININGJOBRESPONSE._serialized_start = 1787
-    _CANCELTRAININGJOBRESPONSE._serialized_end = 1814
-    _MLTRAININGSERVICE._serialized_start = 2213
-    _MLTRAININGSERVICE._serialized_end = 2708
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n#app/mltraining/v1/ml_training.proto\x12\x16viam.app.mltraining.v1\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x17google/rpc/status.proto\x1a\x16tagger/v1/tagger.proto"\x90\x05\n\x18SubmitTrainingJobRequest\x12G\n\ndataset_id\x18\x07 \x01(\tB(\x9a\x84\x9e\x03#bson:"dataset_id" json:"dataset_id"R\tdatasetId\x12[\n\x0forganization_id\x18\x02 \x01(\tB2\x9a\x84\x9e\x03-bson:"organization_id" json:"organization_id"R\x0eorganizationId\x12G\n\nmodel_name\x18\x03 \x01(\tB(\x9a\x84\x9e\x03#bson:"model_name" json:"model_name"R\tmodelName\x12S\n\rmodel_version\x18\x04 \x01(\tB.\x9a\x84\x9e\x03)bson:"model_version" json:"model_version"R\x0cmodelVersion\x12j\n\nmodel_type\x18\x05 \x01(\x0e2!.viam.app.mltraining.v1.ModelTypeB(\x9a\x84\x9e\x03#bson:"model_type" json:"model_type"R\tmodelType\x12\x83\x01\n\x0fmodel_framework\x18\x08 \x01(\x0e2&.viam.app.mltraining.v1.ModelFrameworkB2\x9a\x84\x9e\x03-bson:"model_framework" json:"model_framework"R\x0emodelFramework\x120\n\x04tags\x18\x06 \x03(\tB\x1c\x9a\x84\x9e\x03\x17bson:"tags" json:"tags"R\x04tagsJ\x04\x08\x01\x10\x02R\x06filter"+\n\x19SubmitTrainingJobResponse\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"\x88\x06\n\x1eSubmitCustomTrainingJobRequest\x12G\n\ndataset_id\x18\x01 \x01(\tB(\x9a\x84\x9e\x03#bson:"dataset_id" json:"dataset_id"R\tdatasetId\x12^\n\x10registry_item_id\x18\x02 \x01(\tB4\x9a\x84\x9e\x03/bson:"registry_item_id" json:"registry_item_id"R\x0eregistryItemId\x12r\n\x15registry_item_version\x18\x06 \x01(\tB>\x9a\x84\x9e\x039bson:"registry_item_version" json:"registry_item_version"R\x13registryItemVersion\x12[\n\x0forganization_id\x18\x03 \x01(\tB2\x9a\x84\x9e\x03-bson:"organization_id" json:"organization_id"R\x0eorganizationId\x12G\n\nmodel_name\x18\x04 \x01(\tB(\x9a\x84\x9e\x03#bson:"model_name" json:"model_name"R\tmodelName\x12S\n\rmodel_version\x18\x05 \x01(\tB.\x9a\x84\x9e\x03)bson:"model_version" json:"model_version"R\x0cmodelVersion\x12c\n\targuments\x18\x07 \x03(\x0b2E.viam.app.mltraining.v1.SubmitCustomTrainingJobRequest.ArgumentsEntryR\targuments\x12+\n\x11container_version\x18\x08 \x01(\tR\x10containerVersion\x1a<\n\x0eArgumentsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x028\x01"1\n\x1fSubmitCustomTrainingJobResponse\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"\'\n\x15GetTrainingJobRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"a\n\x16GetTrainingJobResponse\x12G\n\x08metadata\x18\x01 \x01(\x0b2+.viam.app.mltraining.v1.TrainingJobMetadataR\x08metadata"\x82\x01\n\x17ListTrainingJobsRequest\x12\'\n\x0forganization_id\x18\x01 \x01(\tR\x0eorganizationId\x12>\n\x06status\x18\x02 \x01(\x0e2&.viam.app.mltraining.v1.TrainingStatusR\x06status"[\n\x18ListTrainingJobsResponse\x12?\n\x04jobs\x18\x01 \x03(\x0b2+.viam.app.mltraining.v1.TrainingJobMetadataR\x04jobs"\x8c\x10\n\x13TrainingJobMetadata\x123\n\x02id\x18\x07 \x01(\tB#\x9a\x84\x9e\x03\x1ebson:"_id" json:"id,omitempty"R\x02id\x12G\n\ndataset_id\x18\x0b \x01(\tB(\x9a\x84\x9e\x03#bson:"dataset_id" json:"dataset_id"R\tdatasetId\x12[\n\x0forganization_id\x18\x0c \x01(\tB2\x9a\x84\x9e\x03-bson:"organization_id" json:"organization_id"R\x0eorganizationId\x12G\n\nmodel_name\x18\r \x01(\tB(\x9a\x84\x9e\x03#bson:"model_name" json:"model_name"R\tmodelName\x12S\n\rmodel_version\x18\x0e \x01(\tB.\x9a\x84\x9e\x03)bson:"model_version" json:"model_version"R\x0cmodelVersion\x12j\n\nmodel_type\x18\x0f \x01(\x0e2!.viam.app.mltraining.v1.ModelTypeB(\x9a\x84\x9e\x03#bson:"model_type" json:"model_type"R\tmodelType\x12\x83\x01\n\x0fmodel_framework\x18\x11 \x01(\x0e2&.viam.app.mltraining.v1.ModelFrameworkB2\x9a\x84\x9e\x03-bson:"model_framework" json:"model_framework"R\x0emodelFramework\x12R\n\ris_custom_job\x18\x12 \x01(\x08B.\x9a\x84\x9e\x03)bson:"is_custom_job" json:"is_custom_job"R\x0bisCustomJob\x12^\n\x10registry_item_id\x18\x13 \x01(\tB4\x9a\x84\x9e\x03/bson:"registry_item_id" json:"registry_item_id"R\x0eregistryItemId\x12r\n\x15registry_item_version\x18\x14 \x01(\tB>\x9a\x84\x9e\x039bson:"registry_item_version" json:"registry_item_version"R\x13registryItemVersion\x12`\n\x06status\x18\x02 \x01(\x0e2&.viam.app.mltraining.v1.TrainingStatusB \x9a\x84\x9e\x03\x1bbson:"status" json:"status"R\x06status\x12c\n\x0cerror_status\x18\x08 \x01(\x0b2\x12.google.rpc.StatusB,\x9a\x84\x9e\x03\'bson:"error_status" json:"error_status"R\x0berrorStatus\x12c\n\ncreated_on\x18\x03 \x01(\x0b2\x1a.google.protobuf.TimestampB(\x9a\x84\x9e\x03#bson:"created_on" json:"created_on"R\tcreatedOn\x12o\n\rlast_modified\x18\x04 \x01(\x0b2\x1a.google.protobuf.TimestampB.\x9a\x84\x9e\x03)bson:"last_modified" json:"last_modified"R\x0clastModified\x12{\n\x10training_started\x18\t \x01(\x0b2\x1a.google.protobuf.TimestampB4\x9a\x84\x9e\x03/bson:"training_started" json:"training_started"R\x0ftrainingStarted\x12s\n\x0etraining_ended\x18\n \x01(\x0b2\x1a.google.protobuf.TimestampB0\x9a\x84\x9e\x03+bson:"training_ended" json:"training_ended"R\rtrainingEnded\x12Z\n\x0fsynced_model_id\x18\x05 \x01(\tB2\x9a\x84\x9e\x03-bson:"synced_model_id" json:"synced_model_id"R\rsyncedModelId\x120\n\x04tags\x18\x10 \x03(\tB\x1c\x9a\x84\x9e\x03\x17bson:"tags" json:"tags"R\x04tags\x12\x80\x01\n\targuments\x18\x15 \x03(\x0b2:.viam.app.mltraining.v1.TrainingJobMetadata.ArgumentsEntryB&\x9a\x84\x9e\x03!bson:"arguments" json:"arguments"R\targuments\x12c\n\x11container_version\x18\x16 \x01(\tB6\x9a\x84\x9e\x031bson:"container_version" json:"container_version"R\x10containerVersion\x1a<\n\x0eArgumentsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x028\x01J\x04\x08\x01\x10\x02J\x04\x08\x06\x10\x07R\x07requestR\nuser_email"*\n\x18CancelTrainingJobRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"\x1b\n\x19CancelTrainingJobResponse"3\n!DeleteCompletedTrainingJobRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"$\n"DeleteCompletedTrainingJobResponse"u\n\x13TrainingJobLogEntry\x12\x14\n\x05level\x18\x01 \x01(\tR\x05level\x12.\n\x04time\x18\x02 \x01(\x0b2\x1a.google.protobuf.TimestampR\x04time\x12\x18\n\x07message\x18\x03 \x01(\tR\x07message"^\n\x19GetTrainingJobLogsRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12"\n\npage_token\x18\x02 \x01(\tH\x00R\tpageToken\x88\x01\x01B\r\n\x0b_page_token"\x85\x01\n\x1aGetTrainingJobLogsResponse\x12?\n\x04logs\x18\x01 \x03(\x0b2+.viam.app.mltraining.v1.TrainingJobLogEntryR\x04logs\x12&\n\x0fnext_page_token\x18\x02 \x01(\tR\rnextPageToken" \n\x1eListSupportedContainersRequest"\xf5\x01\n\x1fListSupportedContainersResponse\x12n\n\rcontainer_map\x18\x01 \x03(\x0b2I.viam.app.mltraining.v1.ListSupportedContainersResponse.ContainerMapEntryR\x0ccontainerMap\x1ab\n\x11ContainerMapEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x127\n\x05value\x18\x02 \x01(\x0b2!.viam.app.mltraining.v1.ContainerR\x05value:\x028\x01"\x9d\x01\n\tContainer\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x10\n\x03uri\x18\x02 \x01(\tR\x03uri\x12\x1c\n\tframework\x18\x03 \x01(\tR\tframework\x12 \n\x0bdescription\x18\x04 \x01(\tR\x0bdescription\x12,\n\x03eol\x18\x05 \x01(\x0b2\x1a.google.protobuf.TimestampR\x03eol*\x9f\x01\n\tModelType\x12\x1a\n\x16MODEL_TYPE_UNSPECIFIED\x10\x00\x12*\n&MODEL_TYPE_SINGLE_LABEL_CLASSIFICATION\x10\x01\x12)\n%MODEL_TYPE_MULTI_LABEL_CLASSIFICATION\x10\x02\x12\x1f\n\x1bMODEL_TYPE_OBJECT_DETECTION\x10\x03*\xa4\x01\n\x0eModelFramework\x12\x1f\n\x1bMODEL_FRAMEWORK_UNSPECIFIED\x10\x00\x12\x1a\n\x16MODEL_FRAMEWORK_TFLITE\x10\x01\x12\x1e\n\x1aMODEL_FRAMEWORK_TENSORFLOW\x10\x02\x12\x1b\n\x17MODEL_FRAMEWORK_PYTORCH\x10\x03\x12\x18\n\x14MODEL_FRAMEWORK_ONNX\x10\x04*\xe7\x01\n\x0eTrainingStatus\x12\x1f\n\x1bTRAINING_STATUS_UNSPECIFIED\x10\x00\x12\x1b\n\x17TRAINING_STATUS_PENDING\x10\x01\x12\x1f\n\x1bTRAINING_STATUS_IN_PROGRESS\x10\x02\x12\x1d\n\x19TRAINING_STATUS_COMPLETED\x10\x03\x12\x1a\n\x16TRAINING_STATUS_FAILED\x10\x04\x12\x1c\n\x18TRAINING_STATUS_CANCELED\x10\x05\x12\x1d\n\x19TRAINING_STATUS_CANCELING\x10\x062\x9c\x08\n\x11MLTrainingService\x12x\n\x11SubmitTrainingJob\x120.viam.app.mltraining.v1.SubmitTrainingJobRequest\x1a1.viam.app.mltraining.v1.SubmitTrainingJobResponse\x12\x8a\x01\n\x17SubmitCustomTrainingJob\x126.viam.app.mltraining.v1.SubmitCustomTrainingJobRequest\x1a7.viam.app.mltraining.v1.SubmitCustomTrainingJobResponse\x12o\n\x0eGetTrainingJob\x12-.viam.app.mltraining.v1.GetTrainingJobRequest\x1a..viam.app.mltraining.v1.GetTrainingJobResponse\x12u\n\x10ListTrainingJobs\x12/.viam.app.mltraining.v1.ListTrainingJobsRequest\x1a0.viam.app.mltraining.v1.ListTrainingJobsResponse\x12x\n\x11CancelTrainingJob\x120.viam.app.mltraining.v1.CancelTrainingJobRequest\x1a1.viam.app.mltraining.v1.CancelTrainingJobResponse\x12\x93\x01\n\x1aDeleteCompletedTrainingJob\x129.viam.app.mltraining.v1.DeleteCompletedTrainingJobRequest\x1a:.viam.app.mltraining.v1.DeleteCompletedTrainingJobResponse\x12{\n\x12GetTrainingJobLogs\x121.viam.app.mltraining.v1.GetTrainingJobLogsRequest\x1a2.viam.app.mltraining.v1.GetTrainingJobLogsResponse\x12\x8a\x01\n\x17ListSupportedContainers\x126.viam.app.mltraining.v1.ListSupportedContainersRequest\x1a7.viam.app.mltraining.v1.ListSupportedContainersResponseB#Z!go.viam.com/api/app/mltraining/v1b\x06proto3')
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'app.mltraining.v1.ml_training_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+    _globals['DESCRIPTOR']._loaded_options = None
+    _globals['DESCRIPTOR']._serialized_options = b'Z!go.viam.com/api/app/mltraining/v1'
+    _globals['_SUBMITTRAININGJOBREQUEST'].fields_by_name['dataset_id']._loaded_options = None
+    _globals['_SUBMITTRAININGJOBREQUEST'].fields_by_name['dataset_id']._serialized_options = b'\x9a\x84\x9e\x03#bson:"dataset_id" json:"dataset_id"'
+    _globals['_SUBMITTRAININGJOBREQUEST'].fields_by_name['organization_id']._loaded_options = None
+    _globals['_SUBMITTRAININGJOBREQUEST'].fields_by_name['organization_id']._serialized_options = b'\x9a\x84\x9e\x03-bson:"organization_id" json:"organization_id"'
+    _globals['_SUBMITTRAININGJOBREQUEST'].fields_by_name['model_name']._loaded_options = None
+    _globals['_SUBMITTRAININGJOBREQUEST'].fields_by_name['model_name']._serialized_options = b'\x9a\x84\x9e\x03#bson:"model_name" json:"model_name"'
+    _globals['_SUBMITTRAININGJOBREQUEST'].fields_by_name['model_version']._loaded_options = None
+    _globals['_SUBMITTRAININGJOBREQUEST'].fields_by_name['model_version']._serialized_options = b'\x9a\x84\x9e\x03)bson:"model_version" json:"model_version"'
+    _globals['_SUBMITTRAININGJOBREQUEST'].fields_by_name['model_type']._loaded_options = None
+    _globals['_SUBMITTRAININGJOBREQUEST'].fields_by_name['model_type']._serialized_options = b'\x9a\x84\x9e\x03#bson:"model_type" json:"model_type"'
+    _globals['_SUBMITTRAININGJOBREQUEST'].fields_by_name['model_framework']._loaded_options = None
+    _globals['_SUBMITTRAININGJOBREQUEST'].fields_by_name['model_framework']._serialized_options = b'\x9a\x84\x9e\x03-bson:"model_framework" json:"model_framework"'
+    _globals['_SUBMITTRAININGJOBREQUEST'].fields_by_name['tags']._loaded_options = None
+    _globals['_SUBMITTRAININGJOBREQUEST'].fields_by_name['tags']._serialized_options = b'\x9a\x84\x9e\x03\x17bson:"tags" json:"tags"'
+    _globals['_SUBMITCUSTOMTRAININGJOBREQUEST_ARGUMENTSENTRY']._loaded_options = None
+    _globals['_SUBMITCUSTOMTRAININGJOBREQUEST_ARGUMENTSENTRY']._serialized_options = b'8\x01'
+    _globals['_SUBMITCUSTOMTRAININGJOBREQUEST'].fields_by_name['dataset_id']._loaded_options = None
+    _globals['_SUBMITCUSTOMTRAININGJOBREQUEST'].fields_by_name['dataset_id']._serialized_options = b'\x9a\x84\x9e\x03#bson:"dataset_id" json:"dataset_id"'
+    _globals['_SUBMITCUSTOMTRAININGJOBREQUEST'].fields_by_name['registry_item_id']._loaded_options = None
+    _globals['_SUBMITCUSTOMTRAININGJOBREQUEST'].fields_by_name['registry_item_id']._serialized_options = b'\x9a\x84\x9e\x03/bson:"registry_item_id" json:"registry_item_id"'
+    _globals['_SUBMITCUSTOMTRAININGJOBREQUEST'].fields_by_name['registry_item_version']._loaded_options = None
+    _globals['_SUBMITCUSTOMTRAININGJOBREQUEST'].fields_by_name['registry_item_version']._serialized_options = b'\x9a\x84\x9e\x039bson:"registry_item_version" json:"registry_item_version"'
+    _globals['_SUBMITCUSTOMTRAININGJOBREQUEST'].fields_by_name['organization_id']._loaded_options = None
+    _globals['_SUBMITCUSTOMTRAININGJOBREQUEST'].fields_by_name['organization_id']._serialized_options = b'\x9a\x84\x9e\x03-bson:"organization_id" json:"organization_id"'
+    _globals['_SUBMITCUSTOMTRAININGJOBREQUEST'].fields_by_name['model_name']._loaded_options = None
+    _globals['_SUBMITCUSTOMTRAININGJOBREQUEST'].fields_by_name['model_name']._serialized_options = b'\x9a\x84\x9e\x03#bson:"model_name" json:"model_name"'
+    _globals['_SUBMITCUSTOMTRAININGJOBREQUEST'].fields_by_name['model_version']._loaded_options = None
+    _globals['_SUBMITCUSTOMTRAININGJOBREQUEST'].fields_by_name['model_version']._serialized_options = b'\x9a\x84\x9e\x03)bson:"model_version" json:"model_version"'
+    _globals['_TRAININGJOBMETADATA_ARGUMENTSENTRY']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA_ARGUMENTSENTRY']._serialized_options = b'8\x01'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['id']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['id']._serialized_options = b'\x9a\x84\x9e\x03\x1ebson:"_id" json:"id,omitempty"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['dataset_id']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['dataset_id']._serialized_options = b'\x9a\x84\x9e\x03#bson:"dataset_id" json:"dataset_id"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['organization_id']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['organization_id']._serialized_options = b'\x9a\x84\x9e\x03-bson:"organization_id" json:"organization_id"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['model_name']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['model_name']._serialized_options = b'\x9a\x84\x9e\x03#bson:"model_name" json:"model_name"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['model_version']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['model_version']._serialized_options = b'\x9a\x84\x9e\x03)bson:"model_version" json:"model_version"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['model_type']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['model_type']._serialized_options = b'\x9a\x84\x9e\x03#bson:"model_type" json:"model_type"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['model_framework']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['model_framework']._serialized_options = b'\x9a\x84\x9e\x03-bson:"model_framework" json:"model_framework"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['is_custom_job']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['is_custom_job']._serialized_options = b'\x9a\x84\x9e\x03)bson:"is_custom_job" json:"is_custom_job"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['registry_item_id']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['registry_item_id']._serialized_options = b'\x9a\x84\x9e\x03/bson:"registry_item_id" json:"registry_item_id"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['registry_item_version']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['registry_item_version']._serialized_options = b'\x9a\x84\x9e\x039bson:"registry_item_version" json:"registry_item_version"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['status']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['status']._serialized_options = b'\x9a\x84\x9e\x03\x1bbson:"status" json:"status"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['error_status']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['error_status']._serialized_options = b'\x9a\x84\x9e\x03\'bson:"error_status" json:"error_status"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['created_on']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['created_on']._serialized_options = b'\x9a\x84\x9e\x03#bson:"created_on" json:"created_on"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['last_modified']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['last_modified']._serialized_options = b'\x9a\x84\x9e\x03)bson:"last_modified" json:"last_modified"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['training_started']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['training_started']._serialized_options = b'\x9a\x84\x9e\x03/bson:"training_started" json:"training_started"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['training_ended']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['training_ended']._serialized_options = b'\x9a\x84\x9e\x03+bson:"training_ended" json:"training_ended"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['synced_model_id']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['synced_model_id']._serialized_options = b'\x9a\x84\x9e\x03-bson:"synced_model_id" json:"synced_model_id"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['tags']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['tags']._serialized_options = b'\x9a\x84\x9e\x03\x17bson:"tags" json:"tags"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['arguments']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['arguments']._serialized_options = b'\x9a\x84\x9e\x03!bson:"arguments" json:"arguments"'
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['container_version']._loaded_options = None
+    _globals['_TRAININGJOBMETADATA'].fields_by_name['container_version']._serialized_options = b'\x9a\x84\x9e\x031bson:"container_version" json:"container_version"'
+    _globals['_LISTSUPPORTEDCONTAINERSRESPONSE_CONTAINERMAPENTRY']._loaded_options = None
+    _globals['_LISTSUPPORTEDCONTAINERSRESPONSE_CONTAINERMAPENTRY']._serialized_options = b'8\x01'
+    _globals['_MODELTYPE']._serialized_start = 5066
+    _globals['_MODELTYPE']._serialized_end = 5225
+    _globals['_MODELFRAMEWORK']._serialized_start = 5228
+    _globals['_MODELFRAMEWORK']._serialized_end = 5392
+    _globals['_TRAININGSTATUS']._serialized_start = 5395
+    _globals['_TRAININGSTATUS']._serialized_end = 5626
+    _globals['_SUBMITTRAININGJOBREQUEST']._serialized_start = 146
+    _globals['_SUBMITTRAININGJOBREQUEST']._serialized_end = 802
+    _globals['_SUBMITTRAININGJOBRESPONSE']._serialized_start = 804
+    _globals['_SUBMITTRAININGJOBRESPONSE']._serialized_end = 847
+    _globals['_SUBMITCUSTOMTRAININGJOBREQUEST']._serialized_start = 850
+    _globals['_SUBMITCUSTOMTRAININGJOBREQUEST']._serialized_end = 1626
+    _globals['_SUBMITCUSTOMTRAININGJOBREQUEST_ARGUMENTSENTRY']._serialized_start = 1566
+    _globals['_SUBMITCUSTOMTRAININGJOBREQUEST_ARGUMENTSENTRY']._serialized_end = 1626
+    _globals['_SUBMITCUSTOMTRAININGJOBRESPONSE']._serialized_start = 1628
+    _globals['_SUBMITCUSTOMTRAININGJOBRESPONSE']._serialized_end = 1677
+    _globals['_GETTRAININGJOBREQUEST']._serialized_start = 1679
+    _globals['_GETTRAININGJOBREQUEST']._serialized_end = 1718
+    _globals['_GETTRAININGJOBRESPONSE']._serialized_start = 1720
+    _globals['_GETTRAININGJOBRESPONSE']._serialized_end = 1817
+    _globals['_LISTTRAININGJOBSREQUEST']._serialized_start = 1820
+    _globals['_LISTTRAININGJOBSREQUEST']._serialized_end = 1950
+    _globals['_LISTTRAININGJOBSRESPONSE']._serialized_start = 1952
+    _globals['_LISTTRAININGJOBSRESPONSE']._serialized_end = 2043
+    _globals['_TRAININGJOBMETADATA']._serialized_start = 2046
+    _globals['_TRAININGJOBMETADATA']._serialized_end = 4106
+    _globals['_TRAININGJOBMETADATA_ARGUMENTSENTRY']._serialized_start = 1566
+    _globals['_TRAININGJOBMETADATA_ARGUMENTSENTRY']._serialized_end = 1626
+    _globals['_CANCELTRAININGJOBREQUEST']._serialized_start = 4108
+    _globals['_CANCELTRAININGJOBREQUEST']._serialized_end = 4150
+    _globals['_CANCELTRAININGJOBRESPONSE']._serialized_start = 4152
+    _globals['_CANCELTRAININGJOBRESPONSE']._serialized_end = 4179
+    _globals['_DELETECOMPLETEDTRAININGJOBREQUEST']._serialized_start = 4181
+    _globals['_DELETECOMPLETEDTRAININGJOBREQUEST']._serialized_end = 4232
+    _globals['_DELETECOMPLETEDTRAININGJOBRESPONSE']._serialized_start = 4234
+    _globals['_DELETECOMPLETEDTRAININGJOBRESPONSE']._serialized_end = 4270
+    _globals['_TRAININGJOBLOGENTRY']._serialized_start = 4272
+    _globals['_TRAININGJOBLOGENTRY']._serialized_end = 4389
+    _globals['_GETTRAININGJOBLOGSREQUEST']._serialized_start = 4391
+    _globals['_GETTRAININGJOBLOGSREQUEST']._serialized_end = 4485
+    _globals['_GETTRAININGJOBLOGSRESPONSE']._serialized_start = 4488
+    _globals['_GETTRAININGJOBLOGSRESPONSE']._serialized_end = 4621
+    _globals['_LISTSUPPORTEDCONTAINERSREQUEST']._serialized_start = 4623
+    _globals['_LISTSUPPORTEDCONTAINERSREQUEST']._serialized_end = 4655
+    _globals['_LISTSUPPORTEDCONTAINERSRESPONSE']._serialized_start = 4658
+    _globals['_LISTSUPPORTEDCONTAINERSRESPONSE']._serialized_end = 4903
+    _globals['_LISTSUPPORTEDCONTAINERSRESPONSE_CONTAINERMAPENTRY']._serialized_start = 4805
+    _globals['_LISTSUPPORTEDCONTAINERSRESPONSE_CONTAINERMAPENTRY']._serialized_end = 4903
+    _globals['_CONTAINER']._serialized_start = 4906
+    _globals['_CONTAINER']._serialized_end = 5063
+    _globals['_MLTRAININGSERVICE']._serialized_start = 5629
+    _globals['_MLTRAININGSERVICE']._serialized_end = 6681