diff --git a/torchx/runner/api.py b/torchx/runner/api.py index 194530372..0bcf0bff3 100644 --- a/torchx/runner/api.py +++ b/torchx/runner/api.py @@ -438,6 +438,7 @@ def status(self, app_handle: AppHandle) -> Optional[AppStatus]: ) if app_status: app_status.ui_url = desc.ui_url + app_status.ji_url = desc.ji_url return app_status def wait( diff --git a/torchx/schedulers/api.py b/torchx/schedulers/api.py index 450ee7c68..fca2cfee8 100644 --- a/torchx/schedulers/api.py +++ b/torchx/schedulers/api.py @@ -59,6 +59,7 @@ class DescribeAppResponse: msg: str = NONE structured_error_msg: str = NONE ui_url: Optional[str] = None + ji_url: Optional[str] = None roles_statuses: List[RoleStatus] = field(default_factory=list) roles: List[Role] = field(default_factory=list) diff --git a/torchx/specs/api.py b/torchx/specs/api.py index 0a482ca15..b0754d664 100644 --- a/torchx/specs/api.py +++ b/torchx/specs/api.py @@ -494,6 +494,7 @@ class AppStatus: msg: str = "" structured_error_msg: str = NONE ui_url: Optional[str] = None + ji_url: Optional[str] = None roles: List[RoleStatus] = field(default_factory=list) def is_terminal(self) -> bool: