Update api.py (pytorch#899)

yikaiMeta · web-flow · commit eb7e3d87b108 · 2024-04-29T16:17:54.000-04:00
diff --git a/torchx/runner/api.py b/torchx/runner/api.py
@@ -187,6 +187,7 @@ def run_component(
                 parent_run_id=parent_run_id,
             )
             handle = self.schedule(dryrun_info)
+            ctx._torchx_event.workspace = workspace
             ctx._torchx_event.scheduler = none_throws(dryrun_info._scheduler)
             ctx._torchx_event.app_image = none_throws(dryrun_info._app).roles[0].image
             ctx._torchx_event.app_id = parse_app_handle(handle)[2]

Original file line number	Diff line number	Diff line change
`@@ -187,6 +187,7 @@ def run_component(`
`187`	`187`	`parent_run_id=parent_run_id,`
`188`	`188`	`)`
`189`	`189`	`handle = self.schedule(dryrun_info)`
	`190`	`+ ctx._torchx_event.workspace = workspace`
`190`	`191`	`ctx._torchx_event.scheduler = none_throws(dryrun_info._scheduler)`
`191`	`192`	`ctx._torchx_event.app_image = none_throws(dryrun_info._app).roles[0].image`
`192`	`193`	`ctx._torchx_event.app_id = parse_app_handle(handle)[2]`