Skip to content

Modify train containers list command to include custom containers - #890

Merged
Lillian Hwang-Geddes (lvhg) merged 20 commits into
mainfrom
APP-17611-list-custom-containers
Sep 10, 2026
Merged

Lillian Hwang-Geddes (lvhg) merged 20 commits into
mainfrom
APP-17611-list-custom-containers

Conversation

@lvhg

@lvhg Lillian Hwang-Geddes (lvhg) commented Aug 24, 2026

Copy link
Copy Markdown
Member

Updates the proto messages for Container and ListSupportedContainers

https://viam.atlassian.net/browse/APP-17611

@github-actions github-actions Bot added the safe to test committer is a member of this org label Aug 24, 2026
Comment thread proto/viam/app/mltraining/v1/ml_training.proto Outdated
string framework = 3; // unset for org-registered
string description = 4; // will serve as display name
google.protobuf.Timestamp eol = 5; // unset for org-registered
// 6 to 8 are new fields, 7 is set only for org-registered containers:

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Do you mean 6 is set only for org-registered?

Also, I'd make sure any comments in here are clear/helpful, as they end up as a part of documentation (ex. "6 to 8 are new fields" is not helpful, especially longterm)

Copy link
Copy Markdown
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Good point, I've made the comments a bit more clear

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Should be good to go after this fix!

google.protobuf.Timestamp eol = 5; // unset for org-registered
string organization_id = 6;
google.protobuf.Timestamp created_on = 7; // unset for Viam containers
String id = 8;

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lowercase 'string'!

Comment thread proto/viam/app/mltraining/v1/ml_training.proto Outdated
string description = 4; // will serve as display name
google.protobuf.Timestamp eol = 5; // unset for org-registered
string organization_id = 6;
google.protobuf.Timestamp created_on = 7; // unset for Viam containers

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Why? Does it matter if this is included?

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

google.protobuf.Timestamp eol = 5;
string key = 1; // org-registered keys are derived from the Docker container info
string uri = 2;
string framework = 3; // unset for org-registered

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Will users know what org-registered means?

Copy link
Copy Markdown
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

org registered and custom containers are the same thing, maybe I should make that clearer in the comments?

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Maybe stick to "customer container" wording

// 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
// org_id: "xyz"

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is this supposed to include all of the fields from Container?

}

message ListSupportedContainersResponse {
map<string, Container> container_map = 1;

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I know not a part of the PR, but why is this a map? The endpoint is List, which I would expect to return a list

Copy link
Copy Markdown
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'm not sure about this since I didn't write that part. I assume there's some part that converts it to a list. If I had to guess why they made it a map here, probably to make it easier to link the containers with their keys

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Only have the one comment - wondering why created_on does not apply to Viam (public) containers

string description = 4; // will serve as display name
google.protobuf.Timestamp eol = 5; // unset for org-registered
string organization_id = 6;
google.protobuf.Timestamp created_on = 7; // unset for Viam containers

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Comment thread proto/viam/app/mltraining/v1/ml_training.proto Outdated
Co-authored-by: Katie Peters <katiepeters@users.noreply.github.com>
Comment on lines +188 to +189
// This response takes a full Container, but not all the fields will be dispayed by CLI:
// only name/key, uri, eol, description, framework, created_on and visibility

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'd remove this

@lvhg Lillian Hwang-Geddes (lvhg) added the ready-for-protos add this when you want protos to compile on every commit label Sep 10, 2026
@lvhg Lillian Hwang-Geddes (lvhg) removed the ready-for-protos add this when you want protos to compile on every commit label Sep 10, 2026
@lvhg
Lillian Hwang-Geddes (lvhg) merged commit 737e65d into main Sep 10, 2026
3 checks passed
@lvhg
Lillian Hwang-Geddes (lvhg) deleted the APP-17611-list-custom-containers branch September 10, 2026 15:46
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

protos-compiled safe to test committer is a member of this org

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants