Skip to content

Commit 329f880

Browse files
viambotviambot
authored
Automated Protos Update (#1038)
1 parent 2969616 commit 329f880

File tree

9 files changed

+215
-73
lines changed

9 files changed

+215
-73
lines changed

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

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,12 @@ async def DeleteCompletedTrainingJob(self, stream: 'grpclib.server.Stream[app.ml
4040
async def GetTrainingJobLogs(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsRequest, app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsResponse]') -> None:
4141
pass
4242

43+
@abc.abstractmethod
44+
async def ListSupportedContainers(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.ListSupportedContainersRequest, app.mltraining.v1.ml_training_pb2.ListSupportedContainersResponse]') -> None:
45+
pass
46+
4347
def __mapping__(self) -> typing.Dict[str, grpclib.const.Handler]:
44-
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)}
48+
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)}
4549

4650
class UnimplementedMLTrainingServiceBase(MLTrainingServiceBase):
4751

@@ -66,6 +70,9 @@ async def DeleteCompletedTrainingJob(self, stream: 'grpclib.server.Stream[app.ml
6670
async def GetTrainingJobLogs(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsRequest, app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsResponse]') -> None:
6771
raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)
6872

73+
async def ListSupportedContainers(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.ListSupportedContainersRequest, app.mltraining.v1.ml_training_pb2.ListSupportedContainersResponse]') -> None:
74+
raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)
75+
6976
class MLTrainingServiceStub:
7077

7178
def __init__(self, channel: grpclib.client.Channel) -> None:
@@ -75,4 +82,5 @@ def __init__(self, channel: grpclib.client.Channel) -> None:
7582
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)
7683
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)
7784
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)
78-
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)
85+
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)
86+
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)

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

Lines changed: 52 additions & 40 deletions
Large diffs are not rendered by default.

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

Lines changed: 89 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -148,21 +148,23 @@ class SubmitCustomTrainingJobRequest(google.protobuf.message.Message):
148148
MODEL_NAME_FIELD_NUMBER: builtins.int
149149
MODEL_VERSION_FIELD_NUMBER: builtins.int
150150
ARGUMENTS_FIELD_NUMBER: builtins.int
151+
CONTAINER_VERSION_FIELD_NUMBER: builtins.int
151152
dataset_id: builtins.str
152153
registry_item_id: builtins.str
153154
registry_item_version: builtins.str
154155
organization_id: builtins.str
155156
model_name: builtins.str
156157
model_version: builtins.str
158+
container_version: builtins.str
157159

158160
@property
159161
def arguments(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
160162
...
161163

162-
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=...) -> None:
164+
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:
163165
...
164166

165-
def ClearField(self, field_name: typing.Literal['arguments', b'arguments', '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:
167+
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:
166168
...
167169
global___SubmitCustomTrainingJobRequest = SubmitCustomTrainingJobRequest
168170

@@ -278,6 +280,7 @@ class TrainingJobMetadata(google.protobuf.message.Message):
278280
SYNCED_MODEL_ID_FIELD_NUMBER: builtins.int
279281
TAGS_FIELD_NUMBER: builtins.int
280282
ARGUMENTS_FIELD_NUMBER: builtins.int
283+
CONTAINER_VERSION_FIELD_NUMBER: builtins.int
281284
id: builtins.str
282285
dataset_id: builtins.str
283286
organization_id: builtins.str
@@ -290,6 +293,7 @@ class TrainingJobMetadata(google.protobuf.message.Message):
290293
registry_item_version: builtins.str
291294
status: global___TrainingStatus.ValueType
292295
synced_model_id: builtins.str
296+
container_version: builtins.str
293297

294298
@property
295299
def error_status(self) -> google.rpc.status_pb2.Status:
@@ -319,13 +323,13 @@ class TrainingJobMetadata(google.protobuf.message.Message):
319323
def arguments(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
320324
...
321325

322-
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=...) -> None:
326+
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:
323327
...
324328

325329
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:
326330
...
327331

328-
def ClearField(self, field_name: typing.Literal['arguments', b'arguments', '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:
332+
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:
329333
...
330334
global___TrainingJobMetadata = TrainingJobMetadata
331335

@@ -431,4 +435,84 @@ class GetTrainingJobLogsResponse(google.protobuf.message.Message):
431435

432436
def ClearField(self, field_name: typing.Literal['logs', b'logs', 'next_page_token', b'next_page_token']) -> None:
433437
...
434-
global___GetTrainingJobLogsResponse = GetTrainingJobLogsResponse
438+
global___GetTrainingJobLogsResponse = GetTrainingJobLogsResponse
439+
440+
@typing.final
441+
class ListSupportedContainersRequest(google.protobuf.message.Message):
442+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
443+
444+
def __init__(self) -> None:
445+
...
446+
global___ListSupportedContainersRequest = ListSupportedContainersRequest
447+
448+
@typing.final
449+
class ListSupportedContainersResponse(google.protobuf.message.Message):
450+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
451+
452+
@typing.final
453+
class ContainerMapEntry(google.protobuf.message.Message):
454+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
455+
KEY_FIELD_NUMBER: builtins.int
456+
VALUE_FIELD_NUMBER: builtins.int
457+
key: builtins.str
458+
459+
@property
460+
def value(self) -> global___Container:
461+
...
462+
463+
def __init__(self, *, key: builtins.str=..., value: global___Container | None=...) -> None:
464+
...
465+
466+
def HasField(self, field_name: typing.Literal['value', b'value']) -> builtins.bool:
467+
...
468+
469+
def ClearField(self, field_name: typing.Literal['key', b'key', 'value', b'value']) -> None:
470+
...
471+
CONTAINER_MAP_FIELD_NUMBER: builtins.int
472+
473+
@property
474+
def container_map(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___Container]:
475+
"""ex value:
476+
container_map: {
477+
"tf:2.15": Container {
478+
key: "tf:2.15"
479+
uri: "us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-15.py310:latest"
480+
description: "Tensorflow 2.15"
481+
eol: { seconds: 1772630400, nanos: 0 } // 2026-03-03T00:00:00Z
482+
}
483+
}
484+
"""
485+
486+
def __init__(self, *, container_map: collections.abc.Mapping[builtins.str, global___Container] | None=...) -> None:
487+
...
488+
489+
def ClearField(self, field_name: typing.Literal['container_map', b'container_map']) -> None:
490+
...
491+
global___ListSupportedContainersResponse = ListSupportedContainersResponse
492+
493+
@typing.final
494+
class Container(google.protobuf.message.Message):
495+
DESCRIPTOR: google.protobuf.descriptor.Descriptor
496+
KEY_FIELD_NUMBER: builtins.int
497+
URI_FIELD_NUMBER: builtins.int
498+
FRAMEWORK_FIELD_NUMBER: builtins.int
499+
DESCRIPTION_FIELD_NUMBER: builtins.int
500+
EOL_FIELD_NUMBER: builtins.int
501+
key: builtins.str
502+
uri: builtins.str
503+
framework: builtins.str
504+
description: builtins.str
505+
506+
@property
507+
def eol(self) -> google.protobuf.timestamp_pb2.Timestamp:
508+
...
509+
510+
def __init__(self, *, key: builtins.str=..., uri: builtins.str=..., framework: builtins.str=..., description: builtins.str=..., eol: google.protobuf.timestamp_pb2.Timestamp | None=...) -> None:
511+
...
512+
513+
def HasField(self, field_name: typing.Literal['eol', b'eol']) -> builtins.bool:
514+
...
515+
516+
def ClearField(self, field_name: typing.Literal['description', b'description', 'eol', b'eol', 'framework', b'framework', 'key', b'key', 'uri', b'uri']) -> None:
517+
...
518+
global___Container = Container

‎src/viam/gen/common/v1/common_pb2.pyi‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ global___Orientation = Orientation
122122

123123
@typing.final
124124
class PoseInFrame(google.protobuf.message.Message):
125-
"""PoseInFrame contains a pose and the and the reference frame in which it was observed"""
125+
"""PoseInFrame contains a pose and the reference frame in which it was observed"""
126126
DESCRIPTOR: google.protobuf.descriptor.Descriptor
127127
REFERENCE_FRAME_FIELD_NUMBER: builtins.int
128128
POSE_FIELD_NUMBER: builtins.int

0 commit comments

Comments
 (0)