Skip to content

Commit 37c492c

Browse files
yonromaiclaude
andcommitted
[iris] Fix dev_tpu.py to use provider_bundle API
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent d3b7c64 commit 37c492c

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)