xpk 0.17.2__py3-none-any.whl → 0.17.3__py3-none-any.whl

xpk/commands/cluster.py

@@ -49,7 +49,6 @@ from ..core.gcloud_context import (
     zone_to_region,
 )
 from ..core.jobset import update_jobset_resources_if_necessary
-from ..core.kjob import apply_kjob_crds, prepare_kjob, verify_kjob_installed
 from ..core.kueue_manager import (KueueConfig, KueueManager)
 from ..core.nap import enable_autoprovisioning_on_cluster
 from ..core.network import (
@@ -98,7 +97,6 @@ def cluster_adapt(args) -> None:
   if should_validate_dependencies(args):
     validate_dependencies_list([
         SystemDependency.KUBECTL,
-        SystemDependency.KJOB,
         SystemDependency.GCLOUD,
     ])
   args.enable_pathways = False
@@ -188,7 +186,6 @@ def cluster_adapt(args) -> None:
   if install_kueue_code != 0:
     xpk_exit(install_kueue_code)
 
-  install_kjob(args)
   if system.accelerator_type == AcceleratorType.GPU:
     prepare_gpus(system)
 
@@ -308,7 +305,6 @@ def cluster_create(args) -> None:
   if should_validate_dependencies(args):
     validate_dependencies_list([
         SystemDependency.KUBECTL,
-        SystemDependency.KJOB,
         SystemDependency.GCLOUD,
     ])
 
@@ -455,8 +451,6 @@ def cluster_create(args) -> None:
   if install_kueue_code != 0:
     xpk_exit(install_kueue_code)
 
-  install_kjob(args)
-
   if system.accelerator_type == AcceleratorType.GPU:
     prepare_gpus(system)
 
@@ -1343,22 +1337,6 @@ def install_storage_csis(args):
       xpk_exit(update_cluster_command_code)
 
 
-def install_kjob(args):
-  xpk_print('Verifying kjob installation')
-  err_code = verify_kjob_installed()
-  if err_code > 0:
-    xpk_exit(err_code)
-
-  xpk_print('Applying kjob CDRs')
-  err_code = apply_kjob_crds()
-  if err_code > 0:
-    xpk_exit(err_code)
-
-  err_code = prepare_kjob(args)
-  if err_code > 0:
-    xpk_exit(err_code)
-
-
 def _install_kueue(
     args,
     system: SystemCharacteristics,

xpk/commands/cluster_gcluster.py

@@ -38,7 +38,6 @@ from ..core.commands import run_command_for_value
 from ..core.docker_manager import DockerManager
 from ..core.gcloud_context import zone_to_region
 from ..core.gcluster_manager import GclusterManager
-from ..core.kjob import apply_kjob_crds, prepare_kjob
 from ..core.remote_state.fuse_remote_state import FuseStateClient
 from ..core.remote_state.remote_state_client import RemoteStateClient
 from ..utils.console import xpk_exit, xpk_print
@@ -112,18 +111,7 @@ def cluster_create(
   get_cluster_credentials(args)
 
   err_code = __install_kueue(args)
-  if err_code > 0:
-    xpk_exit(err_code)
-
-  err_code = apply_kjob_crds()
-  if err_code > 0:
-    xpk_exit(err_code)
-
-  err_code = prepare_kjob(args)
-  if err_code > 0:
-    xpk_exit(err_code)
-
-  xpk_exit(0)
+  xpk_exit(err_code)
 
 
 def __install_kueue(args) -> int:

xpk/commands/cluster_gcluster_test.py

@@ -46,8 +46,6 @@ def mock_cluster_create_deps(request):
   """Mocks dependencies for cluster_create."""
   with (
       patch("xpk.commands.cluster_gcluster.xpk_exit") as mock_exit,
-      patch("xpk.commands.cluster_gcluster.prepare_kjob") as mock_prep_kjob,
-      patch("xpk.commands.cluster_gcluster.apply_kjob_crds") as mock_apply_kjob,
       patch(
           "xpk.commands.cluster_gcluster.get_cluster_credentials"
       ) as mock_get_creds,
@@ -68,8 +66,6 @@ def mock_cluster_create_deps(request):
   ):
     yield {
         "xpk_exit": mock_exit,
-        "prepare_kjob": mock_prep_kjob,
-        "apply_kjob_crds": mock_apply_kjob,
         "get_cluster_credentials": mock_get_creds,
         "generate_blueprint": mock_gen_bp,
         "prepare_gcluster_manager": mock_prep_gcm,
@@ -85,9 +81,6 @@ def test_install_kueue_standard(
     mock_get_total_chips, mock_args, mock_cluster_create_deps
 ):
   """Tests __install_kueue for a standard installation."""
-  mock_cluster_create_deps["prepare_kjob"].return_value = 0
-  mock_cluster_create_deps["apply_kjob_crds"].return_value = 0
-
   mock_system = SystemCharacteristics(
       topology="N/A",
       vms_per_slice=1,
@@ -138,9 +131,6 @@ def test_install_kueue_with_autoprovisioning(
     mock_enable_autoprovisioning, mock_args, mock_cluster_create_deps
 ):
   """Tests __install_kueue with autoprovisioning enabled."""
-  mock_cluster_create_deps["prepare_kjob"].return_value = 0
-  mock_cluster_create_deps["apply_kjob_crds"].return_value = 0
-
   mock_args.enable_autoprovisioning = True
   mock_system = SystemCharacteristics(
       topology="N/A",

xpk/commands/cluster_test.py

@@ -56,7 +56,6 @@ class _ClusterCreateMocks:
   create_cluster_configmaps: MagicMock
   set_jobset_on_cluster: MagicMock
   get_cluster_location: MagicMock
-  install_kjob: MagicMock
   xpk_exit: MagicMock
   update_jobset_resources_if_necessary: MagicMock
   _install_kueue: MagicMock
@@ -204,9 +203,6 @@ def cluster_create_mocks(mocker) -> _ClusterCreateMocks:
           'xpk.commands.cluster.get_cluster_location',
           return_value='us-central1',
       ),
-      install_kjob=mocker.patch(
-          'xpk.commands.cluster.install_kjob', return_value=0
-      ),
       xpk_exit=mocker.patch('xpk.commands.cluster.xpk_exit'),
       update_jobset_resources_if_necessary=mocker.patch(
           'xpk.commands.cluster.update_jobset_resources_if_necessary',

xpk/commands/kind.py

@@ -20,11 +20,6 @@ from ..core.commands import (
     run_command_with_updates,
 )
 from ..core.cluster import set_jobset_on_cluster, setup_k8s_env
-from ..core.kjob import (
-    verify_kjob_installed,
-    prepare_kjob,
-    apply_kjob_crds,
-)
 from ..core.scheduling import get_total_chips_requested_from_args
 from ..core.storage import install_storage_crd
 from ..core.system_characteristics import (
@@ -48,7 +43,6 @@ def cluster_create(args) -> None:
   if should_validate_dependencies(args):
     validate_dependencies_list([
         SystemDependency.KUBECTL,
-        SystemDependency.KJOB,
         SystemDependency.GCLOUD,
     ])
   xpk_print(f'Starting cluster create for cluster {args.cluster}:', flush=True)
@@ -69,21 +63,6 @@ def cluster_create(args) -> None:
   if set_jobset_on_cluster_code != 0:
     xpk_exit(set_jobset_on_cluster_code)
 
-  xpk_print('Verifying kjob installation')
-  err_code = verify_kjob_installed()
-  if err_code > 0:
-    xpk_exit(err_code)
-
-  xpk_print('Applying kjob CDRs')
-  err_code = apply_kjob_crds()
-  if err_code > 0:
-    xpk_exit(err_code)
-
-  args.kind_cluster = True
-  err_code = prepare_kjob(args)
-  if err_code > 0:
-    xpk_exit(err_code)
-
   k8s_client = setup_k8s_env(args)
   install_storage_crd(k8s_client)
 

xpk/core/config.py

@@ -53,14 +53,6 @@ PROJECT_KEY = 'project-id'
 CLIENT_ID_KEY = 'client-id'
 SEND_TELEMETRY_KEY = 'send-telemetry'
 ZONE_KEY = 'zone'
-KJOB_BATCH_IMAGE = 'batch-image'
-KJOB_BATCH_WORKING_DIRECTORY = 'batch-working-directory'
-KJOB_SHELL_IMAGE = 'shell-image'
-KJOB_SHELL_INTERACTIVE_COMMAND = 'shell-interactive-command'
-KJOB_SHELL_WORKING_DIRECTORY = 'shell-working-directory'
-CONFIGS_KEY = 'configs'
-GKE_ENDPOINT_KEY = 'gke-endpoint'
-DEPENDENCIES_KEY = 'deps-verified-version'
 
 DEFAULT_KEYS = [
     CFG_BUCKET_KEY,
@@ -69,13 +61,6 @@ DEFAULT_KEYS = [
     CLIENT_ID_KEY,
     SEND_TELEMETRY_KEY,
     ZONE_KEY,
-    GKE_ENDPOINT_KEY,
-    DEPENDENCIES_KEY,
-    KJOB_BATCH_IMAGE,
-    KJOB_BATCH_WORKING_DIRECTORY,
-    KJOB_SHELL_IMAGE,
-    KJOB_SHELL_INTERACTIVE_COMMAND,
-    KJOB_SHELL_WORKING_DIRECTORY,
 ]
 VERTEX_TENSORBOARD_FEATURE_FLAG = XPK_CURRENT_VERSION >= '0.4.0'
 

xpk/core/system_characteristics.py

@@ -80,15 +80,6 @@ class GpuConfig:
 
   requires_topology: bool
   gpu_direct_name: Literal['fastrak', 'rdma', 'tcpx', 'tcpxo'] = 'fastrak'
-  kjob_decorator_fn: Optional[Callable[[dict], dict]] = None
-  """A function to decorate the kjob template for GPU-specific configurations.
-
-  Args:
-    job_manifest (dict): The kjob manifest as a dictionary.
-
-  Returns:
-    dict: The modified kjob manifest as a dictionary.
-  """
   nccl_installer: Optional[str] = None
   jobset_decorator_fn: Optional[Callable[[str, list[str]], str]] = None
   """A function to decorate the jobset for GPU-specific configurations.
@@ -106,7 +97,7 @@ class GpuConfig:
     parts = []
     for f in dataclasses.fields(self):
       value = getattr(self, f.name)
-      if f.name in ('kjob_decorator_fn', 'jobset_decorator_fn') and value:
+      if f.name in ('jobset_decorator_fn') and value:
         parts.append(f'{f.name}=<function {value.__name__}>')
       else:
         parts.append(f'{f.name}={repr(value)}')
@@ -420,7 +411,6 @@ UserFacingNameToSystemCharacteristics = {
         gpu_config=GpuConfig(
             requires_topology=True,
             nccl_installer=INSTALLER_NCCL_RDMA_A4X,
-            kjob_decorator_fn=rdma_decorator.decorate_kjob_template,
             jobset_decorator_fn=rdma_decorator.decorate_jobset,
             gpu_direct_name='rdma',
         ),
@@ -439,7 +429,6 @@ UserFacingNameToSystemCharacteristics = {
         gpu_config=GpuConfig(
             requires_topology=True,
             nccl_installer=INSTALLER_NCCL_RDMA_A4X,
-            kjob_decorator_fn=rdma_decorator.decorate_kjob_template,
             jobset_decorator_fn=rdma_decorator.decorate_jobset,
             gpu_direct_name='rdma',
         ),
@@ -458,7 +447,6 @@ UserFacingNameToSystemCharacteristics = {
         gpu_config=GpuConfig(
             requires_topology=True,
             nccl_installer=INSTALLER_NCCL_RDMA,
-            kjob_decorator_fn=rdma_decorator.decorate_kjob_template,
             jobset_decorator_fn=rdma_decorator.decorate_jobset,
             gpu_direct_name='rdma',
         ),
@@ -477,7 +465,6 @@ UserFacingNameToSystemCharacteristics = {
         gpu_config=GpuConfig(
             requires_topology=True,
             nccl_installer=INSTALLER_NCCL_RDMA,
-            kjob_decorator_fn=rdma_decorator.decorate_kjob_template,
             jobset_decorator_fn=rdma_decorator.decorate_jobset,
             gpu_direct_name='rdma',
         ),
@@ -497,7 +484,6 @@ UserFacingNameToSystemCharacteristics = {
         gpu_config=GpuConfig(
             requires_topology=True,
             nccl_installer=INSTALLER_NCCL_TCPX,
-            kjob_decorator_fn=tcpx_decorator.decorate_kjob_template,
             jobset_decorator_fn=tcpx_decorator.decorate_jobset,
             gpu_direct_name='tcpx',
         ),
@@ -517,7 +503,6 @@ UserFacingNameToSystemCharacteristics = {
         gpu_config=GpuConfig(
             requires_topology=True,
             nccl_installer=INSTALLER_NCCL_TCPXO,
-            kjob_decorator_fn=tcpxo_decorator.decorate_kjob_template,
             jobset_decorator_fn=tcpxo_decorator.decorate_jobset,
             gpu_direct_name='tcpxo',
         ),

xpk/core/workload_decorators/rdma_decorator.py

@@ -18,21 +18,6 @@ import yaml
 from ...utils.yaml import literal_string
 
 
-def decorate_kjob_template(job_manifest: dict) -> dict:
-  spec = (
-      job_manifest.setdefault('spec', {})
-      .setdefault('template', {})
-      .setdefault('spec', {})
-  )
-  spec.setdefault('tolerations', [])
-  spec.setdefault('volumes', [])
-
-  add_volumes(job_manifest)
-  add_tolerations(job_manifest)
-  update_gpu_containers(job_manifest)
-  return job_manifest
-
-
 def decorate_jobset(jobset_manifest_str: str, sub_networks: list[str]) -> str:
   """
   Decorates a JobSet manifest with the necessary components for rdma-daemon.

xpk/core/workload_decorators/tcpx_decorator.py

@@ -22,14 +22,6 @@ from ...utils.yaml import literal_string
 tcpx = 'v2.0.11'
 
 
-def decorate_kjob_template(job_manifest: dict) -> dict:
-  add_volumes(job_manifest)
-  add_tolerations(job_manifest)
-  add_tcpx_daemon_container(job_manifest)
-  update_gpu_containers(job_manifest)
-  return job_manifest
-
-
 def decorate_job(job_manifest: dict) -> dict:
   add_annotations(job_manifest)
   add_volumes(job_manifest)

xpk/core/workload_decorators/tcpx_decorator_test.py

@@ -47,24 +47,6 @@ spec:
                 image: my-sidecar-image
 """
 
-# Minimal kjob template for testing
-BASE_KJOB_TEMPLATE = {
-    "spec": {
-        "template": {
-            "spec": {
-                "containers": [
-                    {
-                        "name": "main-gpu-container",
-                        "image": "my-gpu-image",
-                        "resources": {"limits": {"nvidia.com/gpu": 8}},
-                    },
-                    {"name": "sidecar-container", "image": "my-sidecar-image"},
-                ]
-            }
-        }
-    }
-}
-
 # Minimal job manifest for testing
 BASE_JOB_MANIFEST = {
     "spec": {
@@ -205,63 +187,3 @@ def test_decorate_job():
   assert "devices.gke.io/container.tcpx-daemon" in annotations
   assert "networking.gke.io/default-interface" in annotations
   assert "networking.gke.io/interfaces" in annotations
-
-
-def test_decorate_kjob_template():
-  """Tests decorate_kjob_template."""
-  kjob_template = copy.deepcopy(BASE_KJOB_TEMPLATE)
-
-  decorated_manifest = tcpx_decorator.decorate_kjob_template(kjob_template)
-
-  pod_template_spec = decorated_manifest["spec"]["template"]["spec"]
-
-  # Check annotations are NOT added
-  assert "annotations" not in decorated_manifest["spec"]["template"].get(
-      "metadata", {}
-  )
-
-  # Check tolerations
-  tolerations = pod_template_spec["tolerations"]
-  assert {
-      "key": "user-workload",
-      "operator": "Equal",
-      "value": "true",
-      "effect": "NoSchedule",
-  } in tolerations
-
-  # Check volumes
-  volumes = pod_template_spec["volumes"]
-  volume_names = {v["name"] for v in volumes}
-  assert "libraries" in volume_names
-  assert "sys" in volume_names
-  assert "proc-sys" in volume_names
-  assert "tcpx-socket" in volume_names
-  assert "dshm" in volume_names
-
-  # Check init container
-  init_containers = pod_template_spec["initContainers"]
-  assert len(init_containers) == 1
-  tcpx_daemon = init_containers[0]
-  assert tcpx_daemon["name"] == "tcpx-daemon"
-  assert tcpx_daemon["image"].endswith(f":{tcpx_decorator.tcpx}")
-
-  # Check GPU container update
-  gpu_container = pod_template_spec["containers"][0]
-  assert gpu_container["name"] == "main-gpu-container"
-
-  # Check env
-  env_vars = {e["name"]: e["value"] for e in gpu_container["env"]}
-  assert env_vars["LD_LIBRARY_PATH"] == "/usr/local/nvidia/lib64"
-
-  # Check volume mounts
-  volume_mounts = {
-      vm["name"]: vm["mountPath"] for vm in gpu_container["volumeMounts"]
-  }
-  assert volume_mounts["tcpx-socket"] == "/tmp"
-  assert volume_mounts["libraries"] == "/usr/local/nvidia/lib64"
-  assert volume_mounts["dshm"] == "/dev/shm"
-
-  # Check non-GPU container is not updated
-  sidecar_container = pod_template_spec["containers"][1]
-  assert "env" not in sidecar_container
-  assert "volumeMounts" not in sidecar_container

xpk/core/workload_decorators/tcpxo_decorator.py

@@ -22,22 +22,6 @@ from ...utils.yaml import literal_string
 rxdm = 'v1.0.12'
 
 
-def decorate_kjob_template(job_manifest: dict) -> dict:
-  spec = (
-      job_manifest.setdefault('spec', {})
-      .setdefault('template', {})
-      .setdefault('spec', {})
-  )
-  spec.setdefault('tolerations', [])
-  spec.setdefault('volumes', [])
-
-  add_volumes(job_manifest)
-  add_tolerations(job_manifest)
-  add_tcpxo_daemon_container(job_manifest)
-  update_gpu_containers(job_manifest)
-  return job_manifest
-
-
 def decorate_job(job_manifest: dict, sub_networks: list[str]) -> dict:
   job_manifest.setdefault('spec', {}).setdefault('template', {}).setdefault(
       'metadata', {}

xpk/parser/core.py

@@ -23,13 +23,9 @@ from .cluster import set_cluster_parser
 from .inspector import set_inspector_parser
 from .storage import set_storage_parser
 from .workload import set_workload_parsers
-from .batch import set_batch_parser
-from .job import set_job_parser
 from .info import set_info_parser
 from .kind import set_kind_parser
-from .shell import set_shell_parser
 from .version import set_version_parser
-from .run import set_run_parser
 
 
 def set_parser(parser: argparse.ArgumentParser):
@@ -54,20 +50,10 @@ def set_parser(parser: argparse.ArgumentParser):
       "info",
       help="Commands around listing kueue clusterqueues and localqueues.",
   )
-  batch_parser = xpk_subcommands.add_parser(
-      "batch",
-      help="commands around running batch job",
-  )
-  job_parser = xpk_subcommands.add_parser(
-      "job", help="commands around listing, cancelling and investigating jobs"
-  )
   kind_parser = xpk_subcommands.add_parser(
       "kind",
       help="commands around Kind cluster management",
   )
-  shell_parser = xpk_subcommands.add_parser(
-      "shell", help="Commands around configuring and using interactive shell."
-  )
   version_parser = xpk_subcommands.add_parser(
       "version", help="Command to get xpk version"
   )
@@ -76,11 +62,6 @@ def set_parser(parser: argparse.ArgumentParser):
       "config", help="Commands to set and retrieve values from xpk config."
   )
 
-  run_parser = xpk_subcommands.add_parser(
-      "run",
-      help="Command to run parallel jobs",
-  )
-
   def default_subcommand_function(
       _args,
   ) -> int:  # args is unused, so pylint: disable=invalid-name
@@ -96,14 +77,10 @@ def set_parser(parser: argparse.ArgumentParser):
     parser.print_help()
     cluster_parser.print_help()
     workload_parser.print_help()
-    batch_parser.print_help()
     info_parser.print_help()
-    job_parser.print_help()
-    shell_parser.print_help()
     version_parser.print_help()
     kind_parser.print_help()
     config_parser.print_help()
-    run_parser.print_help()
 
     storage_parser.print_help()
     return 0
@@ -111,25 +88,17 @@ def set_parser(parser: argparse.ArgumentParser):
   parser.set_defaults(func=default_subcommand_function)
   workload_parser.set_defaults(func=default_subcommand_function)
   cluster_parser.set_defaults(func=default_subcommand_function)
-  batch_parser.set_defaults(func=default_subcommand_function)
   info_parser.set_defaults(func=default_subcommand_function)
-  job_parser.set_defaults(func=default_subcommand_function)
   kind_parser.set_defaults(func=default_subcommand_function)
-  shell_parser.set_defaults(func=default_subcommand_function)
   storage_parser.set_defaults(func=default_subcommand_function)
   version_parser.set_defaults(func=default_subcommand_function)
   config_parser.set_defaults(func=default_subcommand_function)
-  run_parser.set_defaults(func=default_subcommand_function)
 
   set_workload_parsers(workload_parser=workload_parser)
   set_cluster_parser(cluster_parser=cluster_parser)
   set_inspector_parser(inspector_parser=inspector_parser)
-  set_batch_parser(batch_parser=batch_parser)
   set_info_parser(info_parser=info_parser)
-  set_job_parser(job_parser=job_parser)
   set_kind_parser(kind_parser=kind_parser)
-  set_shell_parser(shell_parser=shell_parser)
   set_storage_parser(storage_parser=storage_parser)
   set_version_parser(version_parser=version_parser)
   set_config_parsers(config_parser=config_parser)
-  set_run_parser(run_parser=run_parser)

xpk/utils/validation.py

@@ -37,14 +37,6 @@ class SystemDependency(Enum):
           ' to install xpk prerequisites.'
       ),
   )
-  KJOB = _SystemDependency(
-      command='kubectl kjob --help',
-      message=(
-          '`kjobctl` not installed. Please follow'
-          ' https://github.com/AI-Hypercomputer/xpk?tab=readme-ov-file#prerequisites'
-          ' to install xpk prerequisites.'
-      ),
-  )
   GCLOUD = _SystemDependency(
       command='gcloud version',
       message=(

{xpk-0.17.2.dist-info → xpk-0.17.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xpk
-Version: 0.17.2
+Version: 0.17.3
 Summary: xpk helps Cloud developers to orchestrate training jobs on accelerators on GKE.
 Author-email: XPK team <xpk-code-reviewers@google.com>
 License: Apache-2.0

{xpk-0.17.2.dist-info → xpk-0.17.3.dist-info}/RECORD

@@ -20,22 +20,17 @@ xpk/blueprints/a4/config-map.yaml.tftpl,sha256=o6LeGIYUfFGyj3vj-8ztV5ildQ46QZVl7
 xpk/blueprints/a4/nccl-rdma-installer-a4.yaml,sha256=if3WOmNLVGTJIJHU76EWC1FyiIXDTRIXcwo4OsBxarQ,2113
 xpk/blueprints/a4/storage_crd.yaml,sha256=r4WFXnSJJ25EUF-t4Ljfbl-cJoSaiFiZkP8451eTub4,1260
 xpk/commands/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
-xpk/commands/batch.py,sha256=Cj1bDpzPMoPdhaKKrOJJLJ3JzRvJrCMn8huQoHHIZJI,4192
-xpk/commands/cluster.py,sha256=DtMiIYdYsciXldoWqAfxPIxl9Hc9kbYIj2LsdBER0PI,46172
-xpk/commands/cluster_gcluster.py,sha256=x26UqoT8RFX5T9ftQXPEL12HMnMFTi8lret16dnZCms,13970
-xpk/commands/cluster_gcluster_test.py,sha256=UcqTTkrQv-R753AtsQvinwgI2vqI6lMHPPEfHPS5e-4,6655
-xpk/commands/cluster_test.py,sha256=-7EjuOoGSZhdnLBNBNCMKM6laDYy02aPncbSfUYcrUs,24147
+xpk/commands/cluster.py,sha256=5ebvHXe8Bi4haMd1RokJbjP0LemqXxC1EseqOLWIkGw,45659
+xpk/commands/cluster_gcluster.py,sha256=Ig8jLjsiyFgw9U4BBEzDK2diA9m0STKQgz-uUTG_vYE,13731
+xpk/commands/cluster_gcluster_test.py,sha256=s1wwkcdY4LTxmk_Tx5PKdh9pZmEoo1n8XlzybWalc0M,6165
+xpk/commands/cluster_test.py,sha256=aMkwKrhoEuqElME16ztx5lwv4zT0z_xV0L3in1RaW6M,24017
 xpk/commands/common.py,sha256=p43sspD5RfYRj3Se_b-X0s0dbBs1PMI1qtySg6zZKKg,2706
 xpk/commands/config.py,sha256=L_zRpQTxMcSh6rxOT8gG263V6YGqzVoz4UxdWywTFdA,850
 xpk/commands/info.py,sha256=uhv5mPfgg9N-5JhQw4dT2jujL9ZC5kzGA18h9NFfm5A,7429
 xpk/commands/inspector.py,sha256=FPasKtGuEZKNXIQin4AG49clfD4b53NxXpWqBPZIIoE,12955
-xpk/commands/job.py,sha256=rPIfWvgm5mLz7K7YDLK721ZcUcg5OEmYVAPAtRtB5Ag,6718
-xpk/commands/kind.py,sha256=GNqsaoLInifFQ_ZGpbN_3xA8ExyeyOqBMdnoPV-PqYI,7813
-xpk/commands/kjob_common.py,sha256=bRaORiGVjPAdN0T3aRmbcQgXYe-EtjoVKePdWzQ5xU4,1928
+xpk/commands/kind.py,sha256=ck4zaJh9kaNluEdBew9OVj4cZXjUYk_ElycqGLo6f7g,7355
 xpk/commands/managed_ml_diagnostics.py,sha256=87wmFbnYQY-kEpJfPo1Up53xM5P_P5wOlXczxHzxJjQ,6984
 xpk/commands/managed_ml_diagnostics_test.py,sha256=pQ1YUGMGRQFJYTS_1o9YyGUzYdLaBdA84LjbnncaeEo,3828
-xpk/commands/run.py,sha256=D0zgmnGeBLATphYhzQj29EScxrMmAKqPRhP6nfWuYcY,4085
-xpk/commands/shell.py,sha256=mRHMwm3Izzsue4bocekm82Rg_cPUaGMClSlvNzNXQ-o,4467
 xpk/commands/storage.py,sha256=cSTJN9Mjvdsvk_Nk43kVdQFhp89nxWbanDsTOGZCkpQ,10708
 xpk/commands/version.py,sha256=k30rdLP9clUM8eeSwRFhpfzSb1qwcQImTfuC59Ed6CA,771
 xpk/commands/workload.py,sha256=l99NRFLs7pXuaLdn5d-Pid-cZulKpB3FNus-HdNDtZw,31513
@@ -47,7 +42,7 @@ xpk/core/cluster.py,sha256=3nl77I_MgQpBZsZSzsiQ_7IyFRzfLrYNRUL1gsSNhKU,24036
 xpk/core/cluster_private.py,sha256=RLi0C7bV0NEUXl6QKQzvUT0weN9EdqPvjuuOQsNO0DY,6868
 xpk/core/cluster_test.py,sha256=J4Wk7E--ik_IsWWzL_iWGWbx99Ih03m-0bs-uU7gGDg,5853
 xpk/core/commands.py,sha256=at73VJHdZ4rVA8uvW997tNrvnCjP9v6zaw96bU0kd74,10841
-xpk/core/config.py,sha256=L3iPFvzFCpW8IEAvlbkuEHYBYXmRTC0BAaR7I_5_Peo,5146
+xpk/core/config.py,sha256=7U8jI5oZcgV_UnOHSS3huUIlDmPNREM-ml0N1Y9IvGM,4612
 xpk/core/config_test.py,sha256=POSuofK0LFbNNygDAo2fjtKY4NMrRjUFeGcpBh9JOS4,3569
 xpk/core/docker_container.py,sha256=8hqWWNKtjf6dqCFRpfndTMGvN_NS6zhfBr7YuKfh7qo,7626
 xpk/core/docker_image.py,sha256=9vwqbb6Mc3C5ZEOph03WS-EWI5hxMYGGigqzIMkDTjE,6909
@@ -74,7 +69,7 @@ xpk/core/resources.py,sha256=dDsG_LOtcU17p1UKgOYyjdPxbMfqcb7pJ4SjfLDA6Os,9389
 xpk/core/scheduling.py,sha256=RMoei_HUs03rfrEC-HYk7ONzg9BRKwr59-KljCR2TMo,11560
 xpk/core/scheduling_test.py,sha256=iYnzXv_MjN743pa4zYAgRqb-6dB9nVPpLI7JP5S8M2I,14463
 xpk/core/storage.py,sha256=NILvVAcLNMLmp4wKx_TEKbMMF5X1oL-FrQV46PT0_ds,16902
-xpk/core/system_characteristics.py,sha256=Tam8wjUz77E6jAJib-r0GsTBmdjo9uaEkXmIdWuzGO8,32844
+xpk/core/system_characteristics.py,sha256=ZQbTbjaeT3Q12kmobz14U878w3FWnXDCetiLZQlVAdY,32127
 xpk/core/system_characteristics_test.py,sha256=sREN8u8bC0ze_q9hY3v-ZxC7so-_Ox1mt_DkIbUgHJ4,7477
 xpk/core/telemetry.py,sha256=R7IONNl5heMoNcOurfT3I34XJrBEODKVY88ONiDGuqE,7512
 xpk/core/telemetry_test.py,sha256=ll-B1ut9X-por17fpQnNb6hKrfyoZanMWRPbvqWrXss,8261
@@ -99,25 +94,21 @@ xpk/core/testing/__init__.py,sha256=PkV8D9WOtlJHH5AIxsQaKeIBcmupT_Ol_bwJgN6G2I8,
 xpk/core/testing/commands_tester.py,sha256=mQOSFggESeTdzqG4srAPV9ezmoeT90r22K58yAty9sE,4445
 xpk/core/testing/commands_tester_test.py,sha256=NnLWh7TJ9rKtb-DtB-vwkxvCe5wNtvUJ0f6sOa87Ht4,4023
 xpk/core/workload_decorators/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
-xpk/core/workload_decorators/rdma_decorator.py,sha256=isbgPnjdu2AT_Da1nVUIRoGE_qZ7jMDOKCgZOLq5r2A,4006
+xpk/core/workload_decorators/rdma_decorator.py,sha256=02HVA_jSyzlVtSQnQj7aPdK03h7v5YyioBqEen6pbj0,3636
 xpk/core/workload_decorators/storage_decorator.py,sha256=DDYQVO1OKTLhveDOA4V6b2RWr4n0fbwHdnoFFmW7iaQ,2000
-xpk/core/workload_decorators/tcpx_decorator.py,sha256=6yvofTv6_XmRfI-nESZjGYeLmGrza1rWxeJGET0TqXU,6182
-xpk/core/workload_decorators/tcpx_decorator_test.py,sha256=iTBS3X_-VwA2oveNDjscduLtll0VOJyFRCp4xmsjg7w,8515
-xpk/core/workload_decorators/tcpxo_decorator.py,sha256=_nLX7tbnxhnS-xv4Jijd1JOP76V4LpNCfW3Np404Cqw,6537
+xpk/core/workload_decorators/tcpx_decorator.py,sha256=cLOntH2ekBcPeiPW0sU3TRozSCpcTxgxpzncrMbRj44,5962
+xpk/core/workload_decorators/tcpx_decorator_test.py,sha256=BmTWsFoBeLb9xhQh3kpqSiarkYax4bj2wLeZ9GrQzag,6089
+xpk/core/workload_decorators/tcpxo_decorator.py,sha256=5SgL-7aTHclN7rvCGvEOjZoUixBmyjfuhVIUBFmneug,6124
 xpk/parser/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
-xpk/parser/batch.py,sha256=mJU-Cp1yTLje59vD-B1IiBcUeD-ZmEsoeB4xhj9cflc,1406
 xpk/parser/cluster.py,sha256=U2T-Q4yS86PWeFLNfknYWDDzZfubCKqIhqasxKLmErI,31342
 xpk/parser/cluster_test.py,sha256=xzQEC3IeAMpwsbNbHLuaNKxR3iaZcm3z4m3i61G62d4,6581
 xpk/parser/common.py,sha256=sJYGjrn2YgFxelDCYB18s1R8Md8GpDcMQNoAezxDDIs,7257
 xpk/parser/common_test.py,sha256=_6Fm2pUF7h4K0G5qxGabXSYr4ng9ihOzlViE6oLQwQs,1557
 xpk/parser/config.py,sha256=-XnWx9aFsBW4Uzo_hpOMD2ZQ0bdZLvq1ksv83_5jqSM,1633
-xpk/parser/core.py,sha256=VRJerlS92ufoQbG1mZv7B04DAP4qGkBHa4pRXgcbAs0,4761
+xpk/parser/core.py,sha256=P2Dx3AbTlDoWnCCrMhVdr3Fs5FEzYFlmSiugkun1GL0,3623
 xpk/parser/info.py,sha256=UJohxVVWdt9IgUXoPsrVae2DN1BjAVGWrSN2ajrB8RQ,1860
 xpk/parser/inspector.py,sha256=hAPAZ2k9iSJgC1mjnz3rMleInsAQ8PmkyyUKFyBmsgY,1997
-xpk/parser/job.py,sha256=5RdE70rucGfrsn65l7Ho6RmO06mag1S0AO-3saVuXyw,4328
 xpk/parser/kind.py,sha256=sgPCqNVrgmFLcOBEbhlaphwVXxMh_opP9ntCq4KPePE,2682
-xpk/parser/run.py,sha256=oi_ksSyJ8Ooffe2EgoV_ecpmXEmNGVotjpIQH-HjufE,1481
-xpk/parser/shell.py,sha256=VC8p-kz9XjJZW9DXZ-rnv41XnRDRpQRFywHpB5j7tfc,1970
 xpk/parser/storage.py,sha256=0V1d1htsjoa-SuxOX_vNxz2Lg4Nue9CBe_H0bNS2Hv0,10270
 xpk/parser/storage_test.py,sha256=i_F9cuQXHRvUy4RJwbfuuI8ZVpTpkkY96sZ1GZ4dLPw,1494
 xpk/parser/validators.py,sha256=-NBZelvfwZRzjz-YUCreD8EzMLHll8PZM-d-MVm2PG4,1192
@@ -156,13 +147,13 @@ xpk/utils/user_agent.py,sha256=1NMtixC1RIr_MwM5pJ0THQ0x1-fCQA92TFHjWAVZldw,1083
 xpk/utils/user_agent_test.py,sha256=lkv8LqzhlA1gXFVeBzoLwE1_iGnm8G9LzkkElMrIrx0,1774
 xpk/utils/user_input.py,sha256=kMdCcPWdkI31f1mJcMsNGda-xKyKxEerpSLpCqIWYPc,1503
 xpk/utils/user_input_test.py,sha256=xO34jkMoTAk5Cmw7yHTk-7YexzC2UZ6ajihV8lnlAyI,2666
-xpk/utils/validation.py,sha256=irL9579RbvwxiGn1t3zhhPo-0oHgdUPOSYsUuFqsDSM,3039
+xpk/utils/validation.py,sha256=rE9LTkXJT7jIesodFb9pONL7ixhLqiQleyoaz7N39Dw,2765
 xpk/utils/validation_test.py,sha256=PEDSMUqZdt_Lx1FSR-LOTXKKtsJ47JH1fxugM0Gfz6Y,1168
 xpk/utils/versions.py,sha256=_Ep68W70a9605XjiaOOpBa9Is9jXlsoOiwL8v5Xt-WA,897
 xpk/utils/yaml.py,sha256=j8xuAJ9yAAwnQi6ozwZ-nMnDyDnc3xWkeBZMtSuP4RU,844
-xpk-0.17.2.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-xpk-0.17.2.dist-info/METADATA,sha256=_G5EPL08DVbtGWPXVmHAg_HxH_-op5be3Fx1rWRJiwI,7930
-xpk-0.17.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-xpk-0.17.2.dist-info/entry_points.txt,sha256=mzEtiIesFkT1kmcTUVDA1o3uOhiniX6tIz2wmOlMu1M,38
-xpk-0.17.2.dist-info/top_level.txt,sha256=TQKZWgV7LSElvmunYT9V_627qOMoxq3qYzWAFzKudB8,16
-xpk-0.17.2.dist-info/RECORD,,
+xpk-0.17.3.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+xpk-0.17.3.dist-info/METADATA,sha256=ONK-6JpzJboT0wF60svxA4SKJTSqeQ4KNfgSvZ_kkDY,7930
+xpk-0.17.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+xpk-0.17.3.dist-info/entry_points.txt,sha256=mzEtiIesFkT1kmcTUVDA1o3uOhiniX6tIz2wmOlMu1M,38
+xpk-0.17.3.dist-info/top_level.txt,sha256=TQKZWgV7LSElvmunYT9V_627qOMoxq3qYzWAFzKudB8,16
+xpk-0.17.3.dist-info/RECORD,,