viam-sdk 0.25.2__py3-none-linux_armv7l.whl → 0.62.0__py3-none-linux_armv7l.whl

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

@@ -1,109 +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 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\x1fgoogle/protobuf/timestamp.proto\x1a\x17google/rpc/status.proto\x1a\x16tagger/v1/tagger.proto"\x8a\x04\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\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"\xb8\x04\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"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"\xe6\r\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\x04tagsJ\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*\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\x92\x06\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.DeleteCompletedTrainingJobResponseB#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['dataset_id']._options = None
-    _SUBMITTRAININGJOBREQUEST.fields_by_name['dataset_id']._serialized_options = b'\x9a\x84\x9e\x03#bson:"dataset_id" json:"dataset_id"'
-    _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"'
-    _SUBMITCUSTOMTRAININGJOBREQUEST.fields_by_name['dataset_id']._options = None
-    _SUBMITCUSTOMTRAININGJOBREQUEST.fields_by_name['dataset_id']._serialized_options = b'\x9a\x84\x9e\x03#bson:"dataset_id" json:"dataset_id"'
-    _SUBMITCUSTOMTRAININGJOBREQUEST.fields_by_name['registry_item_id']._options = None
-    _SUBMITCUSTOMTRAININGJOBREQUEST.fields_by_name['registry_item_id']._serialized_options = b'\x9a\x84\x9e\x03/bson:"registry_item_id" json:"registry_item_id"'
-    _SUBMITCUSTOMTRAININGJOBREQUEST.fields_by_name['registry_item_version']._options = None
-    _SUBMITCUSTOMTRAININGJOBREQUEST.fields_by_name['registry_item_version']._serialized_options = b'\x9a\x84\x9e\x039bson:"registry_item_version" json:"registry_item_version"'
-    _SUBMITCUSTOMTRAININGJOBREQUEST.fields_by_name['organization_id']._options = None
-    _SUBMITCUSTOMTRAININGJOBREQUEST.fields_by_name['organization_id']._serialized_options = b'\x9a\x84\x9e\x03-bson:"organization_id" json:"organization_id"'
-    _SUBMITCUSTOMTRAININGJOBREQUEST.fields_by_name['model_name']._options = None
-    _SUBMITCUSTOMTRAININGJOBREQUEST.fields_by_name['model_name']._serialized_options = b'\x9a\x84\x9e\x03#bson:"model_name" json:"model_name"'
-    _SUBMITCUSTOMTRAININGJOBREQUEST.fields_by_name['model_version']._options = None
-    _SUBMITCUSTOMTRAININGJOBREQUEST.fields_by_name['model_version']._serialized_options = b'\x9a\x84\x9e\x03)bson:"model_version" json:"model_version"'
-    _TRAININGJOBMETADATA.fields_by_name['id']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['id']._serialized_options = b'\x9a\x84\x9e\x03\x1ebson:"_id" json:"id,omitempty"'
-    _TRAININGJOBMETADATA.fields_by_name['dataset_id']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['dataset_id']._serialized_options = b'\x9a\x84\x9e\x03#bson:"dataset_id" json:"dataset_id"'
-    _TRAININGJOBMETADATA.fields_by_name['organization_id']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['organization_id']._serialized_options = b'\x9a\x84\x9e\x03-bson:"organization_id" json:"organization_id"'
-    _TRAININGJOBMETADATA.fields_by_name['model_name']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['model_name']._serialized_options = b'\x9a\x84\x9e\x03#bson:"model_name" json:"model_name"'
-    _TRAININGJOBMETADATA.fields_by_name['model_version']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['model_version']._serialized_options = b'\x9a\x84\x9e\x03)bson:"model_version" json:"model_version"'
-    _TRAININGJOBMETADATA.fields_by_name['model_type']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['model_type']._serialized_options = b'\x9a\x84\x9e\x03#bson:"model_type" json:"model_type"'
-    _TRAININGJOBMETADATA.fields_by_name['model_framework']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['model_framework']._serialized_options = b'\x9a\x84\x9e\x03-bson:"model_framework" json:"model_framework"'
-    _TRAININGJOBMETADATA.fields_by_name['is_custom_job']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['is_custom_job']._serialized_options = b'\x9a\x84\x9e\x03)bson:"is_custom_job" json:"is_custom_job"'
-    _TRAININGJOBMETADATA.fields_by_name['registry_item_id']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['registry_item_id']._serialized_options = b'\x9a\x84\x9e\x03/bson:"registry_item_id" json:"registry_item_id"'
-    _TRAININGJOBMETADATA.fields_by_name['registry_item_version']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['registry_item_version']._serialized_options = b'\x9a\x84\x9e\x039bson:"registry_item_version" json:"registry_item_version"'
-    _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['error_status']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['error_status']._serialized_options = b'\x9a\x84\x9e\x03\'bson:"error_status" json:"error_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['training_started']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['training_started']._serialized_options = b'\x9a\x84\x9e\x03/bson:"training_started" json:"training_started"'
-    _TRAININGJOBMETADATA.fields_by_name['training_ended']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['training_ended']._serialized_options = b'\x9a\x84\x9e\x03+bson:"training_ended" json:"training_ended"'
-    _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['tags']._options = None
-    _TRAININGJOBMETADATA.fields_by_name['tags']._serialized_options = b'\x9a\x84\x9e\x03\x17bson:"tags" json:"tags"'
-    _MODELTYPE._serialized_start = 3637
-    _MODELTYPE._serialized_end = 3796
-    _MODELFRAMEWORK._serialized_start = 3799
-    _MODELFRAMEWORK._serialized_end = 3963
-    _TRAININGSTATUS._serialized_start = 3966
-    _TRAININGSTATUS._serialized_end = 4197
-    _SUBMITTRAININGJOBREQUEST._serialized_start = 146
-    _SUBMITTRAININGJOBREQUEST._serialized_end = 668
-    _SUBMITTRAININGJOBRESPONSE._serialized_start = 670
-    _SUBMITTRAININGJOBRESPONSE._serialized_end = 713
-    _SUBMITCUSTOMTRAININGJOBREQUEST._serialized_start = 716
-    _SUBMITCUSTOMTRAININGJOBREQUEST._serialized_end = 1284
-    _SUBMITCUSTOMTRAININGJOBRESPONSE._serialized_start = 1286
-    _SUBMITCUSTOMTRAININGJOBRESPONSE._serialized_end = 1335
-    _GETTRAININGJOBREQUEST._serialized_start = 1337
-    _GETTRAININGJOBREQUEST._serialized_end = 1376
-    _GETTRAININGJOBRESPONSE._serialized_start = 1378
-    _GETTRAININGJOBRESPONSE._serialized_end = 1475
-    _LISTTRAININGJOBSREQUEST._serialized_start = 1478
-    _LISTTRAININGJOBSREQUEST._serialized_end = 1608
-    _LISTTRAININGJOBSRESPONSE._serialized_start = 1610
-    _LISTTRAININGJOBSRESPONSE._serialized_end = 1701
-    _TRAININGJOBMETADATA._serialized_start = 1704
-    _TRAININGJOBMETADATA._serialized_end = 3470
-    _CANCELTRAININGJOBREQUEST._serialized_start = 3472
-    _CANCELTRAININGJOBREQUEST._serialized_end = 3514
-    _CANCELTRAININGJOBRESPONSE._serialized_start = 3516
-    _CANCELTRAININGJOBRESPONSE._serialized_end = 3543
-    _DELETECOMPLETEDTRAININGJOBREQUEST._serialized_start = 3545
-    _DELETECOMPLETEDTRAININGJOBREQUEST._serialized_end = 3596
-    _DELETECOMPLETEDTRAININGJOBRESPONSE._serialized_start = 3598
-    _DELETECOMPLETEDTRAININGJOBRESPONSE._serialized_end = 3634
-    _MLTRAININGSERVICE._serialized_start = 4200
-    _MLTRAININGSERVICE._serialized_end = 4986
+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

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

@@ -91,21 +91,23 @@ class SubmitTrainingJobRequest(google.protobuf.message.Message):
     MODEL_NAME_FIELD_NUMBER: builtins.int
     MODEL_VERSION_FIELD_NUMBER: builtins.int
     MODEL_TYPE_FIELD_NUMBER: builtins.int
+    MODEL_FRAMEWORK_FIELD_NUMBER: builtins.int
     TAGS_FIELD_NUMBER: builtins.int
     dataset_id: builtins.str
     organization_id: builtins.str
     model_name: builtins.str
     model_version: builtins.str
     model_type: global___ModelType.ValueType
+    model_framework: global___ModelFramework.ValueType
 
     @property
     def tags(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
         ...
 
-    def __init__(self, *, dataset_id: builtins.str=..., organization_id: builtins.str=..., model_name: builtins.str=..., model_version: builtins.str=..., model_type: global___ModelType.ValueType=..., tags: collections.abc.Iterable[builtins.str] | None=...) -> None:
+    def __init__(self, *, dataset_id: builtins.str=..., organization_id: builtins.str=..., model_name: builtins.str=..., model_version: builtins.str=..., model_type: global___ModelType.ValueType=..., model_framework: global___ModelFramework.ValueType=..., tags: collections.abc.Iterable[builtins.str] | None=...) -> None:
         ...
 
-    def ClearField(self, field_name: typing.Literal['dataset_id', b'dataset_id', 'model_name', b'model_name', 'model_type', b'model_type', 'model_version', b'model_version', 'organization_id', b'organization_id', 'tags', b'tags']) -> None:
+    def ClearField(self, field_name: typing.Literal['dataset_id', b'dataset_id', 'model_framework', b'model_framework', 'model_name', b'model_name', 'model_type', b'model_type', 'model_version', b'model_version', 'organization_id', b'organization_id', 'tags', b'tags']) -> None:
         ...
 global___SubmitTrainingJobRequest = SubmitTrainingJobRequest
 
@@ -125,23 +127,44 @@ global___SubmitTrainingJobResponse = SubmitTrainingJobResponse
 @typing.final
 class SubmitCustomTrainingJobRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
+    class ArgumentsEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+        value: builtins.str
+
+        def __init__(self, *, key: builtins.str=..., value: builtins.str=...) -> None:
+            ...
+
+        def ClearField(self, field_name: typing.Literal['key', b'key', 'value', b'value']) -> None:
+            ...
     DATASET_ID_FIELD_NUMBER: builtins.int
     REGISTRY_ITEM_ID_FIELD_NUMBER: builtins.int
     REGISTRY_ITEM_VERSION_FIELD_NUMBER: builtins.int
     ORGANIZATION_ID_FIELD_NUMBER: builtins.int
     MODEL_NAME_FIELD_NUMBER: builtins.int
     MODEL_VERSION_FIELD_NUMBER: builtins.int
+    ARGUMENTS_FIELD_NUMBER: builtins.int
+    CONTAINER_VERSION_FIELD_NUMBER: builtins.int
     dataset_id: builtins.str
     registry_item_id: builtins.str
     registry_item_version: builtins.str
     organization_id: builtins.str
     model_name: builtins.str
     model_version: builtins.str
+    container_version: builtins.str
 
-    def __init__(self, *, dataset_id: builtins.str=..., registry_item_id: builtins.str=..., registry_item_version: builtins.str=..., organization_id: builtins.str=..., model_name: builtins.str=..., model_version: builtins.str=...) -> None:
+    @property
+    def arguments(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
         ...
 
-    def ClearField(self, field_name: typing.Literal['dataset_id', b'dataset_id', 'model_name', b'model_name', 'model_version', b'model_version', 'organization_id', b'organization_id', 'registry_item_id', b'registry_item_id', 'registry_item_version', b'registry_item_version']) -> None:
+    def __init__(self, *, dataset_id: builtins.str=..., registry_item_id: builtins.str=..., registry_item_version: builtins.str=..., organization_id: builtins.str=..., model_name: builtins.str=..., model_version: builtins.str=..., arguments: collections.abc.Mapping[builtins.str, builtins.str] | None=..., container_version: builtins.str=...) -> None:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['arguments', b'arguments', 'container_version', b'container_version', 'dataset_id', b'dataset_id', 'model_name', b'model_name', 'model_version', b'model_version', 'organization_id', b'organization_id', 'registry_item_id', b'registry_item_id', 'registry_item_version', b'registry_item_version']) -> None:
         ...
 global___SubmitCustomTrainingJobRequest = SubmitCustomTrainingJobRequest
 
@@ -224,6 +247,20 @@ global___ListTrainingJobsResponse = ListTrainingJobsResponse
 @typing.final
 class TrainingJobMetadata(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
+    class ArgumentsEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+        value: builtins.str
+
+        def __init__(self, *, key: builtins.str=..., value: builtins.str=...) -> None:
+            ...
+
+        def ClearField(self, field_name: typing.Literal['key', b'key', 'value', b'value']) -> None:
+            ...
     ID_FIELD_NUMBER: builtins.int
     DATASET_ID_FIELD_NUMBER: builtins.int
     ORGANIZATION_ID_FIELD_NUMBER: builtins.int
@@ -242,6 +279,8 @@ class TrainingJobMetadata(google.protobuf.message.Message):
     TRAINING_ENDED_FIELD_NUMBER: builtins.int
     SYNCED_MODEL_ID_FIELD_NUMBER: builtins.int
     TAGS_FIELD_NUMBER: builtins.int
+    ARGUMENTS_FIELD_NUMBER: builtins.int
+    CONTAINER_VERSION_FIELD_NUMBER: builtins.int
     id: builtins.str
     dataset_id: builtins.str
     organization_id: builtins.str
@@ -254,6 +293,7 @@ class TrainingJobMetadata(google.protobuf.message.Message):
     registry_item_version: builtins.str
     status: global___TrainingStatus.ValueType
     synced_model_id: builtins.str
+    container_version: builtins.str
 
     @property
     def error_status(self) -> google.rpc.status_pb2.Status:
@@ -279,13 +319,17 @@ class TrainingJobMetadata(google.protobuf.message.Message):
     def tags(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
         ...
 
-    def __init__(self, *, id: builtins.str=..., dataset_id: builtins.str=..., organization_id: builtins.str=..., model_name: builtins.str=..., model_version: builtins.str=..., model_type: global___ModelType.ValueType=..., model_framework: global___ModelFramework.ValueType=..., is_custom_job: builtins.bool=..., registry_item_id: builtins.str=..., registry_item_version: builtins.str=..., status: global___TrainingStatus.ValueType=..., error_status: google.rpc.status_pb2.Status | None=..., created_on: google.protobuf.timestamp_pb2.Timestamp | None=..., last_modified: google.protobuf.timestamp_pb2.Timestamp | None=..., training_started: google.protobuf.timestamp_pb2.Timestamp | None=..., training_ended: google.protobuf.timestamp_pb2.Timestamp | None=..., synced_model_id: builtins.str=..., tags: collections.abc.Iterable[builtins.str] | None=...) -> None:
+    @property
+    def arguments(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
+        ...
+
+    def __init__(self, *, id: builtins.str=..., dataset_id: builtins.str=..., organization_id: builtins.str=..., model_name: builtins.str=..., model_version: builtins.str=..., model_type: global___ModelType.ValueType=..., model_framework: global___ModelFramework.ValueType=..., is_custom_job: builtins.bool=..., registry_item_id: builtins.str=..., registry_item_version: builtins.str=..., status: global___TrainingStatus.ValueType=..., error_status: google.rpc.status_pb2.Status | None=..., created_on: google.protobuf.timestamp_pb2.Timestamp | None=..., last_modified: google.protobuf.timestamp_pb2.Timestamp | None=..., training_started: google.protobuf.timestamp_pb2.Timestamp | None=..., training_ended: google.protobuf.timestamp_pb2.Timestamp | None=..., synced_model_id: builtins.str=..., tags: collections.abc.Iterable[builtins.str] | None=..., arguments: collections.abc.Mapping[builtins.str, builtins.str] | None=..., container_version: builtins.str=...) -> None:
         ...
 
     def HasField(self, field_name: typing.Literal['created_on', b'created_on', 'error_status', b'error_status', 'last_modified', b'last_modified', 'training_ended', b'training_ended', 'training_started', b'training_started']) -> builtins.bool:
         ...
 
-    def ClearField(self, field_name: typing.Literal['created_on', b'created_on', 'dataset_id', b'dataset_id', 'error_status', b'error_status', 'id', b'id', 'is_custom_job', b'is_custom_job', 'last_modified', b'last_modified', 'model_framework', b'model_framework', 'model_name', b'model_name', 'model_type', b'model_type', 'model_version', b'model_version', 'organization_id', b'organization_id', 'registry_item_id', b'registry_item_id', 'registry_item_version', b'registry_item_version', 'status', b'status', 'synced_model_id', b'synced_model_id', 'tags', b'tags', 'training_ended', b'training_ended', 'training_started', b'training_started']) -> None:
+    def ClearField(self, field_name: typing.Literal['arguments', b'arguments', 'container_version', b'container_version', 'created_on', b'created_on', 'dataset_id', b'dataset_id', 'error_status', b'error_status', 'id', b'id', 'is_custom_job', b'is_custom_job', 'last_modified', b'last_modified', 'model_framework', b'model_framework', 'model_name', b'model_name', 'model_type', b'model_type', 'model_version', b'model_version', 'organization_id', b'organization_id', 'registry_item_id', b'registry_item_id', 'registry_item_version', b'registry_item_version', 'status', b'status', 'synced_model_id', b'synced_model_id', 'tags', b'tags', 'training_ended', b'training_ended', 'training_started', b'training_started']) -> None:
         ...
 global___TrainingJobMetadata = TrainingJobMetadata
 
@@ -329,4 +373,146 @@ class DeleteCompletedTrainingJobResponse(google.protobuf.message.Message):
 
     def __init__(self) -> None:
         ...
-global___DeleteCompletedTrainingJobResponse = DeleteCompletedTrainingJobResponse
+global___DeleteCompletedTrainingJobResponse = DeleteCompletedTrainingJobResponse
+
+@typing.final
+class TrainingJobLogEntry(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    LEVEL_FIELD_NUMBER: builtins.int
+    TIME_FIELD_NUMBER: builtins.int
+    MESSAGE_FIELD_NUMBER: builtins.int
+    level: builtins.str
+    message: builtins.str
+
+    @property
+    def time(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        ...
+
+    def __init__(self, *, level: builtins.str=..., time: google.protobuf.timestamp_pb2.Timestamp | None=..., message: builtins.str=...) -> None:
+        ...
+
+    def HasField(self, field_name: typing.Literal['time', b'time']) -> builtins.bool:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['level', b'level', 'message', b'message', 'time', b'time']) -> None:
+        ...
+global___TrainingJobLogEntry = TrainingJobLogEntry
+
+@typing.final
+class GetTrainingJobLogsRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    ID_FIELD_NUMBER: builtins.int
+    PAGE_TOKEN_FIELD_NUMBER: builtins.int
+    id: builtins.str
+    page_token: builtins.str
+
+    def __init__(self, *, id: builtins.str=..., page_token: builtins.str | None=...) -> None:
+        ...
+
+    def HasField(self, field_name: typing.Literal['_page_token', b'_page_token', 'page_token', b'page_token']) -> builtins.bool:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['_page_token', b'_page_token', 'id', b'id', 'page_token', b'page_token']) -> None:
+        ...
+
+    def WhichOneof(self, oneof_group: typing.Literal['_page_token', b'_page_token']) -> typing.Literal['page_token'] | None:
+        ...
+global___GetTrainingJobLogsRequest = GetTrainingJobLogsRequest
+
+@typing.final
+class GetTrainingJobLogsResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    LOGS_FIELD_NUMBER: builtins.int
+    NEXT_PAGE_TOKEN_FIELD_NUMBER: builtins.int
+    next_page_token: builtins.str
+
+    @property
+    def logs(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TrainingJobLogEntry]:
+        ...
+
+    def __init__(self, *, logs: collections.abc.Iterable[global___TrainingJobLogEntry] | None=..., next_page_token: builtins.str=...) -> None:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['logs', b'logs', 'next_page_token', b'next_page_token']) -> None:
+        ...
+global___GetTrainingJobLogsResponse = GetTrainingJobLogsResponse
+
+@typing.final
+class ListSupportedContainersRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(self) -> None:
+        ...
+global___ListSupportedContainersRequest = ListSupportedContainersRequest
+
+@typing.final
+class ListSupportedContainersResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
+    class ContainerMapEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+
+        @property
+        def value(self) -> global___Container:
+            ...
+
+        def __init__(self, *, key: builtins.str=..., value: global___Container | None=...) -> None:
+            ...
+
+        def HasField(self, field_name: typing.Literal['value', b'value']) -> builtins.bool:
+            ...
+
+        def ClearField(self, field_name: typing.Literal['key', b'key', 'value', b'value']) -> None:
+            ...
+    CONTAINER_MAP_FIELD_NUMBER: builtins.int
+
+    @property
+    def container_map(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___Container]:
+        """ex value:
+        container_map: {
+          "tf:2.15": Container {
+            key: "tf:2.15"
+            uri: "us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-15.py310:latest"
+            description: "Tensorflow 2.15"
+            eol: { seconds: 1772630400, nanos: 0 } // 2026-03-03T00:00:00Z
+          }
+        }
+        """
+
+    def __init__(self, *, container_map: collections.abc.Mapping[builtins.str, global___Container] | None=...) -> None:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['container_map', b'container_map']) -> None:
+        ...
+global___ListSupportedContainersResponse = ListSupportedContainersResponse
+
+@typing.final
+class Container(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    KEY_FIELD_NUMBER: builtins.int
+    URI_FIELD_NUMBER: builtins.int
+    FRAMEWORK_FIELD_NUMBER: builtins.int
+    DESCRIPTION_FIELD_NUMBER: builtins.int
+    EOL_FIELD_NUMBER: builtins.int
+    key: builtins.str
+    uri: builtins.str
+    framework: builtins.str
+    description: builtins.str
+
+    @property
+    def eol(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        ...
+
+    def __init__(self, *, key: builtins.str=..., uri: builtins.str=..., framework: builtins.str=..., description: builtins.str=..., eol: google.protobuf.timestamp_pb2.Timestamp | None=...) -> None:
+        ...
+
+    def HasField(self, field_name: typing.Literal['eol', b'eol']) -> builtins.bool:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['description', b'description', 'eol', b'eol', 'framework', b'framework', 'key', b'key', 'uri', b'uri']) -> None:
+        ...
+global___Container = Container