feat(platform): add RTX_PRO_6000 to the GPUType enum - #720
Open
oatanas wants to merge 1 commit into
Open
Conversation
The he-tme application accepts RTX PRO 6000 (sm_120) GPUs, but the SDK enum
stopped at L4 and A100, so `validate_run_sdk_metadata` — which `Runs.submit`
calls on every submission — rejected the run before it reached the platform:
pipeline.gpu.gpu_type
Input should be 'L4' or 'A100' [type=enum, input_value='RTX_PRO_6000']
That gate is inside the SDK and `submit` has no opt-out, so no caller could
work around it; passing the string through from a downstream CLI still failed
here.
Bumps SDK_METADATA_SCHEMA_VERSION to 0.0.7 and regenerates the published run
schema, following 0.0.6, which was bumped for the equivalent change (adding
FLEX_START to the provisioning-mode enum). Adding an enum member widens what
validates, so runs written against 0.0.6 stay valid.
CLI_REFERENCE.md is left alone: it is generated, has drifted from the current
CLI, and `nox -s docs` regenerates it with all extras installed.
Refs HETA-1792
Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01CNyXyfHNFAgv7zUYjUG22P
4 tasks
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Targets
feat/PYSDK-133/integration. Supersedes #719, which was opened againstmainbefore I knew this was the integration target; #719 is closed. Re-applied by hand rather than cherry-picked, since this branch renames the package tosrc/aignostics_sdk/.Summary
Adds
RTX_PRO_6000toplatform.GPUType. Thehe-tmeapplication already accepts RTX PRO 6000 (sm_120) GPUs for HETA-1792, but the SDK enum stopped atL4andA100.This is not only a CLI-help issue:
Runs.submit()callsvalidate_run_sdk_metadata(custom_metadata["sdk"])on every submission, which validatespipeline.gpu.gpu_typeagainst this enum. Verified against the released SDK 1.4.9:Because the gate is inside
submit(), and that method exposes no opt-out, a downstream caller cannot route around it — stargate's "pass unknown GPU types through" change (stargate#378) still failed here. Adding the member is the fix.Schema version
Bumps
SDK_METADATA_SCHEMA_VERSIONto0.0.7and regeneratesdocs/source/_static/sdk_run_custom_metadata_schema_{v0.0.7,latest}.json. This follows the precedent of0.0.6, which was bumped for the equivalent change of addingFLEX_STARTto the provisioning-mode enum. The change only widens what validates, so metadata written against0.0.6remains valid.Deliberately not included
CLI_REFERENCE.md. It is generated and already drifted from the CLI on this branch; regenerating locally rewrites ~745 lines (aTEXT→<str>typer formatting change, plus sections that disappear unless every extra is installed). It belongs in its own chore commit vianox -s docs.ruff checkauto-fixed an import-organization error inapplication/_cli.pyand a blank-line issue inapplication/_service.py; both reproduce on this branch atHEADwithout my change, so I reverted them to keep the diff minimal. Worth a separate cleanup.Test plan
pytest tests/aignostics/platform/sdk_metadata_test.py -m unit— 71 passedruff check --no-fix/ruff format --checkclean on the changed files"enum": ["L4", "A100", "RTX_PRO_6000"]🤖 Generated with Claude Code
https://claude.ai/code/session_01CNyXyfHNFAgv7zUYjUG22P