Allow empty session name for APIs that take it

Differential Revision: D73799333

Pull Request resolved: meta-pytorch#1057

Author: kiukchung

torchx/runner/test/api_test.py

@@ -25,6 +25,7 @@
     AppDef,
     AppHandle,
     AppState,
+    parse_app_handle,
     Resource,
     Role,
     UnknownAppException,
@@ -132,6 +133,32 @@ def test_session_id(self, record_mock: MagicMock) -> None:
                 event = record_mock.call_args_list[i].args[0]
                 self.assertEqual(event.session, CURRENT_SESSION_ID)
 
+    def test_empty_session_id(self, _: MagicMock) -> None:
+        empty_session_name = ""
+        with Runner(
+            empty_session_name,
+            {"local": cast(SchedulerFactory, create_scheduler)},
+        ) as runner:
+            app = AppDef(
+                "__unused_for_test__",
+                roles=[
+                    Role(
+                        name="echo",
+                        image=str(self.tmpdir),
+                        resource=resource.SMALL,
+                        entrypoint="echo",
+                        args=["__unused_for_test__"],
+                    )
+                ],
+            )
+
+            app_handle = runner.run(app, "local", self.cfg)
+
+            scheduler, session_name, app_id = parse_app_handle(app_handle)
+            self.assertEqual(scheduler, "local")
+            self.assertEqual(empty_session_name, session_name)
+            self.assertEqual(app_handle, f"local:///{app_id}")
+
     def test_run(self, record_mock: MagicMock) -> None:
         test_file = self.tmpdir / "test_file"
 

torchx/specs/api.py

@@ -1115,14 +1115,31 @@ def __init__(self, app_handle: "AppHandle") -> None:
 
 def parse_app_handle(app_handle: AppHandle) -> ParsedAppHandle:
     """
-    parses the app handle into ```(scheduler_backend, session_name, and app_id)```
+    Parses the app handle into ```(scheduler_backend, session_name, and app_id)```.
+
+    Example:
+
+    .. doctest::
+
+     assert parse_app_handle("k8s://default/foo_bar") == ("k8s", "default", "foo_bar")
+     assert parse_app_handle("k8s:///foo_bar") == ("k8s", "", "foo_bar")
+
+    Args:
+        app_handle: a URI of the form ``{scheduler}://{session_name}/{app_id}``,
+            where the ``session_name`` is optional. In this case the app handle is
+            of the form ``{scheduler}:///{app_id}`` (notice the triple slashes).
+
+    Returns: A ``Tuple`` of three elements, ``(scheduler, session_name, app_id)``
+        parsed from the app_handle URI str. If the session name is not present then
+        an empty string is returned in its place in the tuple.
+
     """
 
     # parse it manually b/c currently torchx does not
     # define allowed characters nor length for session name and app_id
     import re
 
-    pattern = r"(?P<scheduler_backend>.+)://(?P<session_name>.+)/(?P<app_id>.+)"
+    pattern = r"(?P<scheduler_backend>.+)://(?P<session_name>.*)/(?P<app_id>.+)"
     match = re.match(pattern, app_handle)
     if not match:
         raise MalformedAppHandleException(app_handle)

torchx/specs/test/api_test.py

@@ -377,6 +377,15 @@ def test_parse_malformed_app_handles(self) -> None:
                 with self.assertRaises(MalformedAppHandleException):
                     parse_app_handle(handle)
 
+    def test_parse_app_handle_empty_session_name(self) -> None:
+        # missing session name is not OK but an empty one is
+        app_handle = "local:///my_application_id"
+        handle = parse_app_handle(app_handle)
+
+        self.assertEqual(handle.app_id, "my_application_id")
+        self.assertEqual("local", handle.scheduler_backend)
+        self.assertEqual("", handle.session_name)
+
     def test_parse(self) -> None:
         (scheduler_backend, session_name, app_id) = parse_app_handle(
             "local://my_session/my_app_id_1234"