torchx-nightly 2024.4.25__py3-none-any.whl → 2024.4.27__py3-none-any.whl

torchx/runner/api.py

@@ -177,7 +177,7 @@ class Runner:
             ComponentNotFoundException: if the ``component_path`` is failed to resolve.
         """
 
-        with log_event("run_component") as ctx:
+        with log_event("run_component", workspace=workspace) as ctx:
             dryrun_info = self.dryrun_component(
                 component,
                 component_args,
@@ -237,7 +237,9 @@ class Runner:
             An application handle that is used to call other action APIs on the app.
         """
 
-        with log_event(api="run", runcfg=json.dumps(cfg) if cfg else None) as ctx:
+        with log_event(
+            api="run", runcfg=json.dumps(cfg) if cfg else None, workspace=workspace
+        ) as ctx:
             dryrun_info = self.dryrun(
                 app,
                 scheduler,
@@ -371,7 +373,12 @@ class Runner:
                     role.env[tracker_config_env_var_name(name)] = config
 
         cfg = cfg or dict()
-        with log_event("dryrun", scheduler, runcfg=json.dumps(cfg) if cfg else None):
+        with log_event(
+            "dryrun",
+            scheduler,
+            runcfg=json.dumps(cfg) if cfg else None,
+            workspace=workspace,
+        ):
             sched = self._scheduler(scheduler)
             resolved_cfg = sched.run_opts().resolve(cfg)
             if workspace and isinstance(sched, WorkspaceMixin):

torchx/runner/events/__init__.py

@@ -85,9 +85,15 @@ class log_event:
         app_id: Optional[str] = None,
         app_image: Optional[str] = None,
         runcfg: Optional[str] = None,
+        workspace: Optional[str] = None,
     ) -> None:
         self._torchx_event: TorchxEvent = self._generate_torchx_event(
-            api, scheduler or "", app_id, app_image=app_image, runcfg=runcfg
+            api,
+            scheduler or "",
+            app_id,
+            app_image=app_image,
+            runcfg=runcfg,
+            workspace=workspace,
         )
         self._start_cpu_time_ns = 0
         self._start_wall_time_ns = 0
@@ -124,6 +130,7 @@ class log_event:
         app_image: Optional[str] = None,
         runcfg: Optional[str] = None,
         source: SourceType = SourceType.UNKNOWN,
+        workspace: Optional[str] = None,
     ) -> TorchxEvent:
         return TorchxEvent(
             session=app_id or "",
@@ -133,4 +140,5 @@ class log_event:
             app_image=app_image,
             runcfg=runcfg,
             source=source,
+            workspace=workspace,
         )

torchx/runner/events/api.py

@@ -47,6 +47,7 @@ class TorchxEvent:
     cpu_time_usec: Optional[int] = None
     wall_time_usec: Optional[int] = None
     start_epoch_time_usec: Optional[int] = None
+    workspace: Optional[str] = None
 
     def __str__(self) -> str:
         return self.serialize()

{torchx_nightly-2024.4.25.dist-info → torchx_nightly-2024.4.27.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: torchx-nightly
-Version: 2024.4.25
+Version: 2024.4.27
 Summary: TorchX SDK and Components
 Home-page: https://github.com/pytorch/torchx
 Author: TorchX Devs

{torchx_nightly-2024.4.25.dist-info → torchx_nightly-2024.4.27.dist-info}/RECORD

@@ -55,10 +55,10 @@ torchx/pipelines/kfp/__init__.py,sha256=8iJ8lql_fxwuk9VCYSxXnX6tPL228fB5mDZpOs-k
 torchx/pipelines/kfp/adapter.py,sha256=q9fGRS3-xOYVl-tAjTNlUFLbw22r5IiCz9IP34t1rWU,8959
 torchx/pipelines/kfp/version.py,sha256=mYBxd6bm4MeR34D--xo-JLQ9wHeAl_ZQLwbItCf9tr0,539
 torchx/runner/__init__.py,sha256=x8Sz7s_tLxPgJgvWIhK4ju9BNZU61uBFywGwDY6CqJs,315
-torchx/runner/api.py,sha256=AuP2qGGPotawzBOrsZHw5pQq8uzDTIelp6PiGNe1DAA,27869
+torchx/runner/api.py,sha256=YPgkqifZqfvHy7FK57qVrLwwaZkt3jit7ALHcuuxhxY,28013
 torchx/runner/config.py,sha256=pBNbe51EvsfyZvOdhAKvobq0V9lKtpnDkWUzEgnxiP4,17838
-torchx/runner/events/__init__.py,sha256=3FBI4y9oaLFUs6TtTjJAWmGV24WLP_GkOo9AfhDrfxg,4179
-torchx/runner/events/api.py,sha256=EgXK2wF_qc_zZHFgVHNcSAaPibyjM91DfQVtkXXcC-E,2195
+torchx/runner/events/__init__.py,sha256=9vaNjo39IR2YE-gDAd23PLvJg7i-hOMDV50nXAsYGlI,4376
+torchx/runner/events/api.py,sha256=JiC3ZgHONE_k9vAmur-8cQ2ZdUAPm45LJB8tBUguGIo,2231
 torchx/runner/events/handlers.py,sha256=ThHCIJW21BfBgB7b6ftyjASJmD1KdizpjuTtsyqnvJs,522
 torchx/runtime/__init__.py,sha256=Wxje2BryzeQneFu5r6P9JJiEKG-_C9W1CcZ_JNrKT6g,593
 torchx/runtime/tracking/__init__.py,sha256=dYnAPnrXYREfPXkpHhdOFkcYIODWEbA13PdD-wLQYBo,3055
@@ -112,9 +112,9 @@ torchx/workspace/__init__.py,sha256=FqN8AN4VhR1C_SBY10MggQvNZmyanbbuPuE-JCjkyUY,
 torchx/workspace/api.py,sha256=1heBmPgB-W5Zf9gwViM7NrqvHpZlVYeMN7jpY8Qkytc,5479
 torchx/workspace/dir_workspace.py,sha256=npNW_IjUZm_yS5r-8hrRkH46ndDd9a_eApT64m1S1T4,2268
 torchx/workspace/docker_workspace.py,sha256=PFu2KQNVC-0p2aKJ-W_BKA9ZOmXdCY2ABEkCExp3udQ,10269
-torchx_nightly-2024.4.25.dist-info/LICENSE,sha256=WVHfXhFC0Ia8LTKt_nJVYobdqTJVg_4J3Crrfm2A8KQ,1721
-torchx_nightly-2024.4.25.dist-info/METADATA,sha256=NkEwKxtGkoWMu3SGsu4-C2H3tKELuIG3ng2kZMp66_o,6053
-torchx_nightly-2024.4.25.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-torchx_nightly-2024.4.25.dist-info/entry_points.txt,sha256=3JYZFlX9aWzR-Gs_qsx1zq7mlqbFz6Mi9rQUULW8caI,170
-torchx_nightly-2024.4.25.dist-info/top_level.txt,sha256=pxew3bc2gsiViS0zADs0jb6kC5v8o_Yy_85fhHj_J1A,7
-torchx_nightly-2024.4.25.dist-info/RECORD,,
+torchx_nightly-2024.4.27.dist-info/LICENSE,sha256=WVHfXhFC0Ia8LTKt_nJVYobdqTJVg_4J3Crrfm2A8KQ,1721
+torchx_nightly-2024.4.27.dist-info/METADATA,sha256=vTpt4RBunHNlIACRcH17wDZdb53GxWTFwXoMlpfrU0c,6053
+torchx_nightly-2024.4.27.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+torchx_nightly-2024.4.27.dist-info/entry_points.txt,sha256=3JYZFlX9aWzR-Gs_qsx1zq7mlqbFz6Mi9rQUULW8caI,170
+torchx_nightly-2024.4.27.dist-info/top_level.txt,sha256=pxew3bc2gsiViS0zADs0jb6kC5v8o_Yy_85fhHj_J1A,7
+torchx_nightly-2024.4.27.dist-info/RECORD,,