Skip to content

Add CustomResourceConfig proto for shell-command launchers#625

Open
kmontemayor2-sc wants to merge 3 commits intomainfrom
kmonte/custom-resource-config-pr1-proto
Open

Add CustomResourceConfig proto for shell-command launchers#625
kmontemayor2-sc wants to merge 3 commits intomainfrom
kmonte/custom-resource-config-pr1-proto

Conversation

@kmontemayor2-sc
Copy link
Copy Markdown
Collaborator

Introduces a new oneof arm on TrainerResourceConfig / InferencerResourceConfig that lets callers describe a launcher as a shell command + positional args, instead of a fixed-shape Vertex AI / KFP / local resource config. The proto carries no semantics here — the dispatcher is added in a follow-up PR; this commit only ships the message, regenerated bindings, and the wrapper-property update so downstream code can read wrapper.trainer_config and get a CustomResourceConfig back.

The diff includes a long tail of cosmetic Scala changes outside gigl_resource_config/ because scalapbc regenerates every sibling proto's emitted source whenever any one proto in the same directory changes. Reviewers can scope to CustomResourceConfig.scala and the *ResourceConfig.scala siblings that gain the new oneof case.

Scope of work done

Where is the documentation for this feature?: N/A

Did you add automated tests or write a test plan?

Updated Changelog.md? NO

Ready for code review?: NO

Introduces a new `oneof` arm on `TrainerResourceConfig` /
`InferencerResourceConfig` that lets callers describe a launcher as a
shell command + positional args, instead of a fixed-shape Vertex AI /
KFP / local resource config. The proto carries no semantics here — the
dispatcher is added in a follow-up PR; this commit only ships the
message, regenerated bindings, and the wrapper-property update so
downstream code can read `wrapper.trainer_config` and get a
`CustomResourceConfig` back.

The diff includes a long tail of cosmetic Scala changes outside
`gigl_resource_config/` because scalapbc regenerates every sibling
proto's emitted source whenever any one proto in the same directory
changes. Reviewers can scope to `CustomResourceConfig.scala` and the
`*ResourceConfig.scala` siblings that gain the new oneof case.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
@kmontemayor2-sc
Copy link
Copy Markdown
Collaborator Author

/all_test

@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented May 6, 2026

GiGL Automation

@ 23:07:46UTC : 🔄 Python Unit Test started.

@ 23:14:06UTC : ❌ Workflow failed.
Please check the logs for more details.

@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented May 6, 2026

GiGL Automation

@ 23:07:46UTC : 🔄 C++ Unit Test started.

@ 23:09:45UTC : ✅ Workflow completed successfully.

@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented May 6, 2026

GiGL Automation

@ 23:07:51UTC : 🔄 Integration Test started.

@ 24:29:53UTC : ✅ Workflow completed successfully.

@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented May 6, 2026

GiGL Automation

@ 23:07:52UTC : 🔄 Scala Unit Test started.

@ 23:18:38UTC : ✅ Workflow completed successfully.

@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented May 6, 2026

GiGL Automation

@ 23:07:53UTC : 🔄 E2E Test started.

@ 24:30:49UTC : ✅ Workflow completed successfully.

@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented May 6, 2026

GiGL Automation

@ 23:07:55UTC : 🔄 Lint Test started.

@ 23:24:00UTC : ✅ Workflow completed successfully.

@kmontemayor2-sc kmontemayor2-sc marked this pull request as ready for review May 7, 2026 23:30
@kmontemayor2-sc kmontemayor2-sc requested a review from zfan3-sc as a code owner May 7, 2026 23:30
Copy link
Copy Markdown
Collaborator

@mkolodner-sc mkolodner-sc left a comment

Choose a reason for hiding this comment

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

Stamped

@kmontemayor2-sc kmontemayor2-sc added this pull request to the merge queue May 7, 2026
@github-merge-queue github-merge-queue Bot removed this pull request from the merge queue due to failed status checks May 7, 2026
@kmontemayor2-sc kmontemayor2-sc added this pull request to the merge queue May 7, 2026
@kmontemayor2-sc kmontemayor2-sc removed this pull request from the merge queue due to a manual request May 7, 2026
CustomResourceConfig is launcher-pluggable — there is no concrete
machine spec to validate against (no machine_type, num_workers, GPU
config, etc.). The wrapper's trainer_config / inferencer_config
properties now return a union that includes CustomResourceConfig
(introduced earlier in this PR), but _validate_machine_config does
not accept it.

Add an isinstance early-return guard at each call site that logs
the skip and returns. Shape and backend-compatibility validation
for CustomResourceConfig come in a follow-up PR; this commit only
makes the existing validation flow type-clean against the widened
union.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
@kmontemayor2-sc kmontemayor2-sc force-pushed the kmonte/custom-resource-config-pr1-proto branch from 8c727e3 to 0425547 Compare May 8, 2026 00:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants