Skip to content

Commit 375cc85

Browse files
yonromaiclaude
authored andcommitted
[iris] Fix dev_tpu.py to use provider_bundle API (#4395)
## Summary - `scripts/iris/dev_tpu.py` still called `IrisConfig.platform()`, which was removed in #3900 - Replace with `provider_bundle().controller` to match the current `IrisConfig` API Fixes #4394 ## Test plan - [x] `uv run python scripts/iris/dev_tpu.py --help` — CLI loads without import errors - [x] `uv run pytest lib/iris/tests/test_dev_tpu.py` — 4/4 pass - [x] `uv run python scripts/iris/dev_tpu.py --config lib/iris/examples/marin.yaml --tpu-name test-dev-tpu allocate --tpu-type v6e-8` — tunnel established, job submitted to live cluster 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-authored-by: Romain Yon <1596570+yonromai@users.noreply.github.com> Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent 1f0d467 commit 375cc85

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

scripts/iris/dev_tpu.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -169,11 +169,12 @@ def find_workspace_root(start: Path) -> Path | None:
169169
def controller_client(config_file: str) -> Iterable[IrisClient]:
170170
iris_config = IrisConfig.load(config_file)
171171
controller_address = iris_config.controller_address()
172-
platform = iris_config.platform()
172+
providers = iris_config.provider_bundle()
173+
controller = providers.controller
173174
workspace = find_workspace_root(Path.cwd())
174175
if not controller_address:
175-
controller_address = platform.discover_controller(iris_config.proto.controller)
176-
with platform.tunnel(address=controller_address) as tunneled:
176+
controller_address = controller.discover_controller(iris_config.proto.controller)
177+
with controller.tunnel(address=controller_address) as tunneled:
177178
client = IrisClient.remote(tunneled, workspace=workspace)
178179
try:
179180
yield client

0 commit comments

Comments
 (0)