Update api.py (#899)

pytorch · Apr 29, 2024 · eb7e3d8 · eb7e3d8
1 parent 31e281e
commit eb7e3d8
Showing 1 changed file with 1 addition and 0 deletions.
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]