addressed comments

Author: Edwinhr716

axlearn/cloud/gcp/job_test.py

@@ -220,7 +220,7 @@ def run(self, result=None):
         # Run tests under mock user and settings.
         self._settings = default_mock_settings()
         with mock_gcp_settings(
-            [pathways_utils.__name__, bundler.__name__],
+            [jobset_utils.__name__, bundler.__name__],
             settings=self._settings,
         ):
             return super().run(result)
@@ -281,7 +281,9 @@ class Config(Bundler.Config):
         self.assertEqual(cfg.name, cfg.builder.name)
         self.assertEqual(cfg.project, self._settings["project"])
         self.assertEqual(cfg.zone, self._settings["zone"])
-        self.assertEqual(cfg.builder.reservation, reservation or self._settings["gke_reservation"])
+        self.assertEqual(
+            cfg.builder.inner.reservation, reservation or self._settings["gke_reservation"]
+        )
         self.assertEqual(cfg.num_replicas, 1)
         # Should work with wrapped bundlers.
         if wrap_bundler:

axlearn/cloud/gcp/utils.py

@@ -401,7 +401,7 @@ def delete_k8s_leaderworkerset(name: str, *, namespace: str):
         raise
 
 
-def list_k8s_leaderworkerset(name: str, *, namespace: str) -> list[str]:
+def list_k8s_leaderworkerset(*, namespace: str) -> list[str]:
     """List a K8s LWS by name, including all descendant jobs.
 
     Args:
@@ -418,7 +418,6 @@ def list_k8s_leaderworkerset(name: str, *, namespace: str) -> list[str]:
     import kubernetes as k8s  # pytype: disable=import-error
 
     lws_groups = k8s.client.CustomObjectsApi().list_namespaced_custom_object(
-        name=name,
         namespace=namespace,
         propagation_policy="Foreground",
         **custom_leaderworkerset_kwargs(),