xpk 0.11.0__py3-none-any.whl → 0.12.0__py3-none-any.whl

xpk/parser/common.py

@@ -15,24 +15,31 @@ limitations under the License.
 """
 
 import argparse
+from typing import Protocol, Any
+
+
+class ParserOrArgumentGroup(Protocol):
+
+  def add_argument(self, *args, **kwargs) -> Any:
+    ...
 
 
 def add_shared_arguments(
-    custom_parser: argparse.ArgumentParser, required=False
+    custom_parser_or_group: ParserOrArgumentGroup, required=False
 ) -> None:
-  """Add shared arguments to the parser.
+  """Add shared arguments to the parser or argument group.
 
   Args:
-    custom_parser: parser to add shared arguments to.
+    custom_parser_or_group: parser or argument group to add shared arguments to.
   """
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '--project',
       type=str,
       default=None,
       help='GCE project name, defaults to "gcloud config project."',
       required=required,
   )
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '--zone',
       type=str,
       default=None,
@@ -43,7 +50,7 @@ def add_shared_arguments(
       ),
       required=required,
   )
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '--dry-run',
       type=bool,
       action=argparse.BooleanOptionalAction,
@@ -58,14 +65,14 @@ def add_shared_arguments(
 
 
 def add_cluster_arguments(
-    custom_parser: argparse.ArgumentParser, required=False
+    custom_parser_or_group: ParserOrArgumentGroup, required=False
 ) -> None:
-  """Add cluster argument to the parser.
+  """Add cluster argument to the parser or argument group.
 
   Args:
-    custom_parser: parser to add shared arguments to.
+    custom_parser_or_group: parser or argument group to add shared arguments to.
   """
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '--cluster',
       type=str,
       default=None,
@@ -74,13 +81,15 @@ def add_cluster_arguments(
   )
 
 
-def add_kind_cluster_arguments(custom_parser: argparse.ArgumentParser) -> None:
-  """Add kind cluster arguments to the parser.
+def add_kind_cluster_arguments(
+    custom_parser_or_group: ParserOrArgumentGroup,
+) -> None:
+  """Add kind cluster arguments to the parser or argument group.
 
   Args:
-    custom_parser: parser to add shared arguments to.
+    custom_parser_or_group: parser or argument group to add shared arguments to.
   """
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '--kind-cluster',
       type=bool,
       action=argparse.BooleanOptionalAction,
@@ -89,13 +98,13 @@ def add_kind_cluster_arguments(custom_parser: argparse.ArgumentParser) -> None:
   )
 
 
-def add_global_arguments(custom_parser: argparse.ArgumentParser):
+def add_global_arguments(custom_parser_or_group: ParserOrArgumentGroup):
   """Add global - no cloud dependent -  arguments to the parser.
 
   Args:
-    custom_parser: parser to add global arguments to.
+    custom_parser_or_group: parser or argument group to add global arguments to.
   """
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '--dry-run',
       type=bool,
       action=argparse.BooleanOptionalAction,
@@ -108,20 +117,20 @@ def add_global_arguments(custom_parser: argparse.ArgumentParser):
   )
 
 
-def add_slurm_arguments(custom_parser: argparse.ArgumentParser):
+def add_slurm_arguments(custom_parser_or_group: ParserOrArgumentGroup):
   """Add Slurm job arguments to the parser.
 
   Args:
-    custom_parser: parser to add global arguments to.
+    custom_parser_or_group: parser or argument group to add global arguments to.
   """
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '--ignore-unknown-flags',
       type=bool,
       action=argparse.BooleanOptionalAction,
       default=False,
       help='Ignore all the unsupported flags in the bash script.',
   )
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '-a',
       '--array',
       type=str,
@@ -137,32 +146,32 @@ def add_slurm_arguments(custom_parser: argparse.ArgumentParser):
           ' 0. The maximum index value is 2147483647.'
       ),
   )
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '-c',
       '--cpus-per-task',
       type=str,
       default=None,
       help='How much cpus a container inside a pod requires.',
   )
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '--gpus-per-task',
       type=str,
       default=None,
       help='How much gpus a container inside a pod requires.',
   )
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '--mem',
       type=str,
       default=None,
       help='How much memory a pod requires.',
   )
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '--mem-per-task',
       type=str,
       default=None,
       help='How much memory a container requires.',
   )
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '--mem-per-cpu',
       type=str,
       default=None,
@@ -171,7 +180,7 @@ def add_slurm_arguments(custom_parser: argparse.ArgumentParser):
           'of requested cpus per task by mem-per-cpu.'
       ),
   )
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '--mem-per-gpu',
       type=str,
       default=None,
@@ -180,21 +189,21 @@ def add_slurm_arguments(custom_parser: argparse.ArgumentParser):
           'of requested gpus per task by mem-per-gpu.'
       ),
   )
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '-N',
       '--nodes',
       type=int,
       default=None,
       help='Number of pods to be used at a time.',
   )
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '-n',
       '--ntasks',
       type=int,
       default=None,
       help='Number of identical containers inside of a pod, usually 1.',
   )
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '-o',
       '--output',
       type=str,
@@ -204,7 +213,7 @@ def add_slurm_arguments(custom_parser: argparse.ArgumentParser):
           ' passed it proceeds to stdout, and is available via kubectl logs.'
       ),
   )
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '-e',
       '--error',
       type=str,
@@ -214,27 +223,27 @@ def add_slurm_arguments(custom_parser: argparse.ArgumentParser):
           ' proceeds to stdout, and is available via kubectl logs.'
       ),
   )
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '--input',
       type=str,
       default=None,
       help='What to pipe into the script.',
   )
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '-J',
       '--job-name',
       type=str,
       default=None,
       help='What is the job name.',
   )
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '-D',
       '--chdir',
       type=str,
       default=None,
       help='Change directory before executing the script.',
   )
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '-t',
       '--time',
       type=str,
@@ -247,7 +256,7 @@ def add_slurm_arguments(custom_parser: argparse.ArgumentParser):
           'and "days-hours:minutes:seconds".'
       ),
   )
-  custom_parser.add_argument(
+  custom_parser_or_group.add_argument(
       '--priority',
       type=str,
       default='medium',

xpk/parser/storage.py

@@ -28,6 +28,13 @@ from .common import (
     add_kind_cluster_arguments,
     add_shared_arguments,
 )
+from typing import Protocol, Any
+
+
+class Subcommands(Protocol):
+
+  def add_parser(self, *args, **kwargs) -> Any:
+    ...
 
 
 def set_storage_parser(storage_parser: argparse.ArgumentParser) -> None:
@@ -47,7 +54,7 @@ def set_storage_parser(storage_parser: argparse.ArgumentParser) -> None:
 
 
 def add_storage_attach_parser(
-    storage_subcommands_parser: argparse.ArgumentParser,
+    storage_subcommands_parser: Subcommands,
 ) -> None:
 
   storage_attach_parser: argparse.ArgumentParser = (
@@ -171,9 +178,7 @@ def add_storage_attach_parser(
   add_kind_cluster_arguments(opt_args)
 
 
-def add_storage_create_parser(
-    storage_subcommands_parser: argparse.ArgumentParser,
-) -> None:
+def add_storage_create_parser(storage_subcommands_parser: Subcommands) -> None:
   storage_create_parser: argparse.ArgumentParser = (
       storage_subcommands_parser.add_parser(
           'create', help='create XPK Storage.'
@@ -272,9 +277,7 @@ def add_storage_create_parser(
   add_kind_cluster_arguments(opt_args)
 
 
-def add_storage_list_parser(
-    storage_subcommands_parser: argparse.ArgumentParser,
-):
+def add_storage_list_parser(storage_subcommands_parser: Subcommands) -> None:
   storage_list_parser: argparse.ArgumentParser = (
       storage_subcommands_parser.add_parser('list', help='List XPK Storages.')
   )
@@ -290,9 +293,7 @@ def add_storage_list_parser(
   )
 
 
-def add_storage_detach_parser(
-    storage_subcommands_parser: argparse.ArgumentParser,
-):
+def add_storage_detach_parser(storage_subcommands_parser: Subcommands) -> None:
   storage_detach_parser: argparse.ArgumentParser = (
       storage_subcommands_parser.add_parser(
           'detach', help='Detach XPK Storage.'
@@ -315,9 +316,7 @@ def add_storage_detach_parser(
   add_kind_cluster_arguments(opt_args)
 
 
-def add_storage_delete_parser(
-    storage_subcommands_parser: argparse.ArgumentParser,
-):
+def add_storage_delete_parser(storage_subcommands_parser: Subcommands) -> None:
   storage_delete_parser: argparse.ArgumentParser = (
       storage_subcommands_parser.add_parser(
           'delete', help='Delete XPK Storage.'

xpk/parser/workload.py

@@ -193,43 +193,6 @@ def set_workload_parsers(workload_parser):
       ),
   )
 
-  # Autoprovisioning workload arguments
-  workload_create_autoprovisioning_arguments.add_argument(
-      '--on-demand',
-      action='store_true',
-      help=(
-          'Sets autoprovisioning to use on-demand resources for the workload'
-          ' request. See `--reservation` or `--spot` for other capacity types.'
-      ),
-  )
-  workload_create_autoprovisioning_arguments.add_argument(
-      '--reservation',
-      type=str,
-      help=(
-          'Sets autoprovisioning to use reservation resources for the workload'
-          ' request. This will attempt to find the provided reservation. See'
-          ' `--spot`, `--flex` or `--on-demand` for other capacity types.'
-      ),
-  )
-  workload_create_autoprovisioning_arguments.add_argument(
-      '--spot',
-      action='store_true',
-      help=(
-          'Sets autoprovisioning to use spot resources. See `--reservation`,'
-          ' `--flex` or `--on-demand` for other capacity types.'
-      ),
-  )
-
-  workload_create_autoprovisioning_arguments.add_argument(
-      '--flex',
-      action='store_true',
-      help=(
-          'Sets autoprovisioning to use flex-start resources. See'
-          ' `--reservation`, `--spot` or `--on-demand` for other capacity'
-          ' types.'
-      ),
-  )
-
   # "workload create-pathways" command parser.
   workload_create_pathways_parser = workload_subcommands.add_parser(
       'create-pathways', help='Create a new job.'
@@ -257,6 +220,12 @@ def set_workload_parsers(workload_parser):
       '`--base-docker-image` is used by default. Set this argument if the'
       ' user wants the docker image to be used directly by the xpk workload.',
   )
+  workload_create_pathways_autoprovisioning_arguments = (
+      workload_create_pathways_parser.add_argument_group(
+          'Optional Autoprovisioning Arguments',
+          'Arguments for configuring autoprovisioning.',
+      )
+  )
   workload_create_pathways_vertex_tensorboard_arguments = (
       workload_create_pathways_parser.add_argument_group(
           'Vertex Tensorboard Arguments',
@@ -407,6 +376,10 @@ def set_workload_parsers(workload_parser):
       workload_vertex_tensorboard_arguments,
       workload_create_pathways_vertex_tensorboard_arguments,
   ])
+  add_shared_workload_create_autoprovisioning_arguments([
+      workload_create_autoprovisioning_arguments,
+      workload_create_pathways_autoprovisioning_arguments,
+  ])
 
   # Set defaults for both workload create and workload create-pathways after adding all shared args.
   workload_create_parser.set_defaults(func=workload_create)
@@ -452,7 +425,7 @@ def set_workload_parsers(workload_parser):
       type=str,
       help=(
           'Filters the arguments based on job name. Provide a regex'
-          ' expressionto parse jobs that match the pattern or provide a job'
+          ' expression to parse jobs that match the pattern or provide a job'
           ' name to delete a single job.'
       ),
   )
@@ -521,7 +494,7 @@ def set_workload_parsers(workload_parser):
       type=str,
       help=(
           'Filters the arguments based on job name. Provide a regex'
-          ' expressionto parse jobs that match the pattern or provide a job'
+          ' expression to parse jobs that match the pattern or provide a job'
           ' name to view a single job.'
       ),
       required=False,
@@ -770,3 +743,48 @@ def add_shared_workload_create_tensorboard_arguments(args_parsers):
             '<cluster>-<workload> will be created.'
         ),
     )
+
+
+def add_shared_workload_create_autoprovisioning_arguments(args_parsers):
+  """Add shared autoprovisioning arguments
+
+  Args:
+      List of workload create optional arguments parsers
+  """
+  for custom_parser in args_parsers:
+    custom_parser.add_argument(
+        '--on-demand',
+        action='store_true',
+        help=(
+            'Sets autoprovisioning to use on-demand resources for the workload'
+            ' request. See `--reservation` or `--spot` for other capacity'
+            ' types.'
+        ),
+    )
+    custom_parser.add_argument(
+        '--reservation',
+        type=str,
+        help=(
+            'Sets autoprovisioning to use reservation resources for the'
+            ' workload request. This will attempt to find the provided'
+            ' reservation. See `--spot`, `--flex` or `--on-demand` for other'
+            ' capacity types.'
+        ),
+    )
+    custom_parser.add_argument(
+        '--spot',
+        action='store_true',
+        help=(
+            'Sets autoprovisioning to use spot resources. See `--reservation`,'
+            ' `--flex` or `--on-demand` for other capacity types.'
+        ),
+    )
+    custom_parser.add_argument(
+        '--flex',
+        action='store_true',
+        help=(
+            'Sets autoprovisioning to use flex-start resources. See'
+            ' `--reservation`, `--spot` or `--on-demand` for other capacity'
+            ' types.'
+        ),
+    )

xpk/utils/console.py

@@ -15,6 +15,7 @@ limitations under the License.
 """
 
 import sys
+from typing import NoReturn
 
 
 def xpk_print(*args, **kwargs):
@@ -29,7 +30,7 @@ def xpk_print(*args, **kwargs):
   sys.stdout.flush()
 
 
-def xpk_exit(error_code):
+def xpk_exit(error_code) -> NoReturn:
   """Helper function to exit xpk with an associated error code.
 
   Args:

{xpk-0.11.0.dist-info → xpk-0.12.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xpk
-Version: 0.11.0
+Version: 0.12.0
 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
@@ -28,6 +28,9 @@ Requires-Dist: pylint>=2.6.0; extra == "dev"
 Requires-Dist: pre-commit; extra == "dev"
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: docker==7.1.0; extra == "dev"
+Requires-Dist: mypy~=1.17; extra == "dev"
+Requires-Dist: types-PyYAML==6.0.2; extra == "dev"
+Requires-Dist: types-docker~=7.1.0.0; extra == "dev"
 Dynamic: license-file
 
 <!--

{xpk-0.11.0.dist-info → xpk-0.12.0.dist-info}/RECORD

@@ -1,68 +1,73 @@
 xpk/__init__.py,sha256=7mu-VQDQMyxM5To0KOhuYe4y2TYGsEkfV7hXZmUyih4,561
-xpk/main.py,sha256=wFc_kIM7kALGIY-JOcoa8m4BCWNRjl5tQ6ZDpv7HpSU,2350
+xpk/main.py,sha256=fQJ4eZMgmBbfYoHT1_d98PdxSC6Gn9Ug-2REAwL0kys,2365
 xpk/api/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
 xpk/api/storage_crd.yaml,sha256=r4WFXnSJJ25EUF-t4Ljfbl-cJoSaiFiZkP8451eTub4,1260
 xpk/commands/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
 xpk/commands/batch.py,sha256=qzKkXf45brC-Ln3C-lajcwdqCc2BCVbKJe5M06AdnKQ,3806
-xpk/commands/cluster.py,sha256=50sezU0GgJGssK1CdrP_XDaRs2UGFP3q7eEivkfmhkc,40296
-xpk/commands/cluster_gcluster.py,sha256=F2QWBo7IYHcX_4jjMvaJoCXisJvrWmGiRbAtPikDS34,10816
-xpk/commands/common.py,sha256=DLmvQ0cjdiS6rMAopyzNSAKGlLH-XLeW4-FTGOlWZeo,2401
+xpk/commands/cluster.py,sha256=X8G6vqn9-JW7d906qII4Sf-o1wyHEA_XEC7fLv1SmWM,40336
+xpk/commands/cluster_gcluster.py,sha256=8jJ7nHBbkmaPtsVQ2m_GnLxkS5iNV5sSN61KL0K_uEY,10861
+xpk/commands/common.py,sha256=aeZLbnpwwKNbc9SCl1U3RugVLUpiK1MH67YQrurpGnU,2419
 xpk/commands/config.py,sha256=gFNkf3ibsvZmcPpkpKXe-KJmHO5IKucNwLCXNgKvaDc,836
-xpk/commands/info.py,sha256=BHqFFXm3Lg1P8qH1Z3gEXmh141-8udduS5EBk38auDg,7251
+xpk/commands/info.py,sha256=1orA0u5KCB6fj-smHkuFL1WCH96NGrEiDpRCgPrxUW4,7304
 xpk/commands/inspector.py,sha256=bwbZW-cLtiBw2V0zvoMprHWhgMbAYm0ez0GjjEqeUR8,12097
-xpk/commands/job.py,sha256=LCFB_l1v5x_k4Ov15hPDAhadcvMZlqvHkObNNuHMCdo,5479
+xpk/commands/job.py,sha256=pZdj0XWXpDQyvDAHf6JrZQl2heZ8sBzoGzB_aqzHc_k,5599
 xpk/commands/kind.py,sha256=Vl3RT47kHCR0ORX9dK37HCiYtbmXJUCIAaq-QEbIclU,7578
-xpk/commands/kjob_common.py,sha256=XTT8uog4PvlxjH7sFTNnvMTlPSzARVM71wj_Czt4XF0,1926
+xpk/commands/kjob_common.py,sha256=dtT-R0n50byTmu2Qcni0pqKYobUAHNENBN_4pt0l-KE,1952
 xpk/commands/run.py,sha256=5hYMG0DcdHnFWsJ5gmfX09t6ZPVItt7FFoHO_ED0_Dk,3798
 xpk/commands/shell.py,sha256=5-sKcI2Rbk3aCojnBNtipCwgOrbIDnG4f8ah0KIayY8,4182
-xpk/commands/storage.py,sha256=tCH2medguOH5bBtywEuqXRtFeHDFE8sq4YVLuOIx_Hk,10600
+xpk/commands/storage.py,sha256=J3kRam3DDwCbvm6JM9h1Zy7_OV7albzAmHi5VkzqyYg,10612
 xpk/commands/version.py,sha256=CU4mb71r66U28krnPAopC6vBpdK-IGclsy5uNaQcgRY,824
-xpk/commands/workload.py,sha256=nc5vClmCmRU04ItwiC3cldCLyEVjzvDUpxv987fgQ-A,26470
+xpk/commands/workload.py,sha256=00KhL0L5j125gxdVLjKyaqXw74oQs1Cbe6lZXV4-sGg,26699
 xpk/core/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
-xpk/core/capacity.py,sha256=pO7H9pkEQwr4sWJP63EprKQph6DYhbIYDQN_TRkA4TQ,7308
-xpk/core/cluster.py,sha256=UHr6DIREO2RKsenrAXMCsvmFdutgcJxZJqeiJy-c88g,29017
-xpk/core/cluster_private.py,sha256=J2-UZ6t5f-UNdcSkuUr2_f4xk6xyrMH9Muk56trBh4M,6657
-xpk/core/commands.py,sha256=JiS4vJqWSLu8MKFBIKPBea9MKD2ZdpaQrziVQBqiDr4,10719
-xpk/core/config.py,sha256=7VN6aQqI6n7mCG1r9KE5-egbYXwjpRiynrZR1IKaNzo,3475
+xpk/core/capacity.py,sha256=SQzncJSLuI4LLJ2VcnpxcRlTjiBG3e8nM0_QxG1986w,7367
+xpk/core/cluster.py,sha256=8_aci7xmR-poDpbirW0Rlwtv7Q9r0K8dshY6CYlpmTM,28948
+xpk/core/cluster_private.py,sha256=NhjaqBZ5uwn6gtsyh6y9ss7hvKXcaKPmpzJjBGBSvQk,6705
+xpk/core/commands.py,sha256=AOLpvcbJoQzY1LEQ_nrbHb4VX6jjxtV9TB1aphu0peA,10759
+xpk/core/config.py,sha256=cGGpJH1IvRCVuYepDmGeaFy6iLTo-ICC949bC1AHnak,3407
+xpk/core/config_test.py,sha256=v1qfyFRzLkYSQ7Wn4nx1N0dBSOFXidLWDfhkeHDZOVM,1847
 xpk/core/docker_container.py,sha256=GvkCJ2S5UKn8uh3pZhRd3X7iS0-PsQpRO8l7QhywVGc,7604
 xpk/core/docker_image.py,sha256=fEdpLQg1C205mMbLREy48WnhvNv2Nm4KQFX87B2CuiA,6624
-xpk/core/docker_manager.py,sha256=mtOEnzSP1GF28ymHSPlTminujh4wWyg5EbC9UT3VFpI,10566
-xpk/core/docker_resources.py,sha256=h-jT5AK1yL7reGPMeSUqb4Khulimljb16J7xoRI2XFw,12753
-xpk/core/filestore.py,sha256=7M-HAiXR-gEu3BJUgRY3cqEIfjo_FL0BAxq9MljEBt4,8022
-xpk/core/gcloud_context.py,sha256=p_LhWHo7GZonear2oupvTO-DpKqEkL0St7PnfxieRDY,5866
+xpk/core/docker_manager.py,sha256=JBFgyD6O7LKwEHJC7YuSoCDZqrFRtb-LjgWNqkfAbR0,10566
+xpk/core/docker_resources.py,sha256=DLaFi7D72pRUUOMpKoccKf91meVBnWqlHBrJXI31GlY,12718
+xpk/core/filestore.py,sha256=mcuUzsAPARbnrBG4fIGsEoN8NmzjaQ6k0tvIwMtjO9k,8068
+xpk/core/gcloud_context.py,sha256=go0avmBbYx45vk_7W3iwQEphmQUx27oaL6dseyocqLI,5836
 xpk/core/gcluster_manager.py,sha256=JFip2hInFczFP2h5AXa70IPIuTaJ475TG6GxkQjKOI8,6337
 xpk/core/gcsfuse.py,sha256=kg5pgxdTjgiqquuGjev9fXzJPb8oiWPTK6wzCddzheQ,2125
 xpk/core/jobset.py,sha256=L_q67_Gq_XjsZKcnznVlCuk1RGZeDFAGDXdCGxbu1oc,4123
-xpk/core/kjob.py,sha256=bl0lIlfFqJqn9fQ2PO-xG6sRPq8RFgk8-cPL2hs-dZM,14681
-xpk/core/kueue.py,sha256=Pww37KL9pR9XaeoKRQ3gUao2lrB1VfB2ZLKf6Zt4Ff0,15097
+xpk/core/kjob.py,sha256=g44jUbchPOxaOAI5ue7WX24wgJwDPsITBeKXmSMnJo0,14714
+xpk/core/kueue.py,sha256=ulm_m38rZeVsskj6OgNtfBbMRkLYUUo_uktEd795B9U,15247
 xpk/core/monitoring.py,sha256=v9MvLzNfvJAVby_ehSlPe6PaO0_pf3shkXg5gd-UWm8,4338
 xpk/core/mtc.py,sha256=pO7p3l-EzLFdTE8MdwWV8i0Zu-7epGql_kPoksVofIU,6259
-xpk/core/nap.py,sha256=kGluT05HOFW9Kq-cZ13pUVyBkAiyA371NvKTiqdYZAk,12791
+xpk/core/nap.py,sha256=L6hVz1gPp2Hcss-Z_7CDCvX70DNU9LFsIU_ILUQ9npc,12814
 xpk/core/network.py,sha256=hQR5Kab5Q5CYjggriNIhNh78Aq0CCF-vPUQI6BC4Wgw,10537
 xpk/core/nodepool.py,sha256=mHYa51og95hwHIxXuzWEycudfewDogzYULoEwrWWTDE,23029
+xpk/core/nodepool_test.py,sha256=QRpmdyZTPRDE2qCibWeKQgE3Q2WCxXt1Onfv0MK4QZQ,2626
 xpk/core/pathways.py,sha256=g4PUSi6RPqpCPVlziWGj14W7zbdNvkw8mrOq09J3s68,10594
 xpk/core/ray.py,sha256=UxOpIc2enHi1fQ4h3KO8FH8bIyEMtYzGtPoeqJKGG4o,6337
-xpk/core/resources.py,sha256=uezEuHw2OzpM4LT2c2EjUCPr9lhBTfLnOPay7hGVyj4,8276
+xpk/core/resources.py,sha256=itTBXPmy8mv6D2kQkzX9RGT5otr-cxKdhHdQSZvvm7w,8076
 xpk/core/scheduling.py,sha256=HIagFbVAl7lz5gRyjpzwke1tLN4PGEyOkUrtBdmy5Ok,9157
-xpk/core/storage.py,sha256=AjpRQn6zUn7qL3W17LuzBtMqyB-3i9YEap5Vw01cUpQ,16926
-xpk/core/system_characteristics.py,sha256=FZ2FYsabECPML-ROYuJ74X4U64vES19U8CujcDImXh8,15161
+xpk/core/storage.py,sha256=NILvVAcLNMLmp4wKx_TEKbMMF5X1oL-FrQV46PT0_ds,16902
+xpk/core/system_characteristics.py,sha256=2mtQlUiufK98XUXo0_f1D4d06FRGdUk_VNkaBg48Fcs,15152
 xpk/core/vertex.py,sha256=pD9UBL62xHomuqdNu7xKccfD2KCbjgohMk3AhX-CXSw,3644
-xpk/core/workload.py,sha256=nfmL4_2Rr0dt5pctHm89KNJcYzulk6CX5d9QEguthJY,8526
+xpk/core/workload.py,sha256=WyT-H4XdeDY07glL1ikmcs44kLgLjMuMtclyZTxrmg8,8921
+xpk/core/workload_test.py,sha256=tVTvrwDRXD3O1GCoftgEBWilCYTN74ayP1KRP0vptx0,857
 xpk/core/blueprint/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
 xpk/core/blueprint/blueprint_definitions.py,sha256=5i331XA-2yP_ALyB6XU5tP2Tf9iHcIX5g0TilxQi8zE,1800
-xpk/core/blueprint/blueprint_generator.py,sha256=vdxoyQnD2uyi8DaG3QtzdVjizf1nNlwoB23ecHUpkKQ,37967
+xpk/core/blueprint/blueprint_generator.py,sha256=Uoz3YKsZKLfIoPI8d3bQ5RNrFFsqBOWb3HBQr3OolHo,38010
+xpk/core/blueprint/blueprint_test.py,sha256=f3dL7KU9n6M8Hdnrg4VI5EEi9GA2dnwm4LDf9n5Wbuw,7410
 xpk/core/remote_state/__init__.py,sha256=PkV8D9WOtlJHH5AIxsQaKeIBcmupT_Ol_bwJgN6G2I8,561
 xpk/core/remote_state/fuse_remote_state.py,sha256=3Dx4ZZd0NFF5-MlqGWHzz8H4bjYiPOWdF_YSEnKUPQ8,3246
 xpk/core/remote_state/remote_state_client.py,sha256=6PcR92Xy_RMjlF4AscanQ1jXNHnewLWGNC2v53jbzD4,1077
 xpk/core/workload_decorators/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
-xpk/core/workload_decorators/rdma_decorator.py,sha256=GU9QQ-cvosurjnhcXqj6e_TpXVHoZE0Of24lALeWPVc,3972
-xpk/core/workload_decorators/storage_decorator.py,sha256=Bj1lRh65s40AJDsWM0xTiHFaWtKC272eImjIjN8Z38c,1967
-xpk/core/workload_decorators/tcpx_decorator.py,sha256=oB60_9744MfS-7yZ3mOnb1KWDRvsNyCCuR5jkp2H_i0,5863
-xpk/core/workload_decorators/tcpxo_decorator.py,sha256=kV1Mzrz0sISPOIRi8I2yuvBhTEfu4e-EYJuz0A63Ubs,6504
+xpk/core/workload_decorators/rdma_decorator.py,sha256=isbgPnjdu2AT_Da1nVUIRoGE_qZ7jMDOKCgZOLq5r2A,4006
+xpk/core/workload_decorators/storage_decorator.py,sha256=DDYQVO1OKTLhveDOA4V6b2RWr4n0fbwHdnoFFmW7iaQ,2000
+xpk/core/workload_decorators/tcpx_decorator.py,sha256=m5EgzEHjbcOD13ygY91mQdhwQt4Gr5PyalVkKcHyeV8,5975
+xpk/core/workload_decorators/tcpx_decorator_test.py,sha256=iTBS3X_-VwA2oveNDjscduLtll0VOJyFRCp4xmsjg7w,8515
+xpk/core/workload_decorators/tcpxo_decorator.py,sha256=_nLX7tbnxhnS-xv4Jijd1JOP76V4LpNCfW3Np404Cqw,6537
 xpk/parser/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
 xpk/parser/batch.py,sha256=mJU-Cp1yTLje59vD-B1IiBcUeD-ZmEsoeB4xhj9cflc,1406
-xpk/parser/cluster.py,sha256=Zb4kKekItzk1lberAxf3jkUH8pvFqjKC8_bKrIXsK9c,28325
-xpk/parser/common.py,sha256=_F2rwsZka15difkvPA1yPARWr9I9ewx8PMzgwMLTvjM,7220
+xpk/parser/cluster.py,sha256=EpC7Kx8o84DRxqrxI1XMklc5raTX7vZSqeZcBZniopk,29131
+xpk/parser/common.py,sha256=N6P6wPuptluNEddh9kpUsaWgxXGADNZLMfKT-P7QkW0,7791
 xpk/parser/config.py,sha256=-XnWx9aFsBW4Uzo_hpOMD2ZQ0bdZLvq1ksv83_5jqSM,1633
 xpk/parser/core.py,sha256=VRJerlS92ufoQbG1mZv7B04DAP4qGkBHa4pRXgcbAs0,4761
 xpk/parser/info.py,sha256=UJohxVVWdt9IgUXoPsrVae2DN1BjAVGWrSN2ajrB8RQ,1860
@@ -71,14 +76,14 @@ 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=lzGQSgngWnguz6yCNUUOJj65BVB_C_WC6nNksHBeTw4,9914
+xpk/parser/storage.py,sha256=XNynqulEzTmT8_G6wkeBwfXX0XQ1lsd6BFcx0H6rGfU,9971
 xpk/parser/validators.py,sha256=-NBZelvfwZRzjz-YUCreD8EzMLHll8PZM-d-MVm2PG4,1192
 xpk/parser/version.py,sha256=eJo4PAbbmRQZulgKBs_ytbVgV9zAaaXeNzMMxmgFMVY,769
-xpk/parser/workload.py,sha256=UVOVCngwhJ4ikuINsdK4rOFlzr39Sv6q0S49NDuNaI0,25493
+xpk/parser/workload.py,sha256=J0Wg4NEO7pOdOwf_8uC9bVzXodBXwkYo1c2gRJVyyak,26068
 xpk/templates/__init__.py,sha256=7mu-VQDQMyxM5To0KOhuYe4y2TYGsEkfV7hXZmUyih4,561
 xpk/templates/storage.yaml,sha256=AykdyMtDnKZF8Y_0BYxoYP03hEIzEk6iNalXAQHgAls,163
 xpk/utils/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
-xpk/utils/console.py,sha256=bKibWIswcB1aWGZp0ZpL-NEhvTrxJMy7wWD4-3BVTKI,1479
+xpk/utils/console.py,sha256=hRbvtog_VAzuxt5GfwK5GZdd5SWaa7kvWG8zo_qFRQc,1519
 xpk/utils/file.py,sha256=jlv2o4ah9UmWJ7NuOCnTwtMZFLerOATBIMQeQ03-kIw,2142
 xpk/utils/gcs_utils.py,sha256=zg-XSTv4G4TFjeT2bNBm2WLdDXPrOZi0rNv_JdppNg4,4113
 xpk/utils/kubectl.py,sha256=WKB9UhpouPN9G4n2ejRi_PgsYLI0R01gzkS1WGU6mJA,1828
@@ -87,9 +92,9 @@ xpk/utils/objects.py,sha256=OwMNxB4TGX21qnJPdZo2YBMPMbQPqOtHMh19QhoRNRY,2498
 xpk/utils/templates.py,sha256=g8zgR1MxyJmTmzM_wnvH30FmcbgQMC47UQwBtLj8B9k,807
 xpk/utils/validation.py,sha256=bSJApIY0Lk48I4EEQP08ZUvolXt_APpYXVGJXFQ_YLA,2711
 xpk/utils/yaml.py,sha256=j8xuAJ9yAAwnQi6ozwZ-nMnDyDnc3xWkeBZMtSuP4RU,844
-xpk-0.11.0.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-xpk-0.11.0.dist-info/METADATA,sha256=j-s1ky4Hnojze2t84S9u6H6TaeAOWwQrFC8Q4k2aF5Y,71613
-xpk-0.11.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-xpk-0.11.0.dist-info/entry_points.txt,sha256=mzEtiIesFkT1kmcTUVDA1o3uOhiniX6tIz2wmOlMu1M,38
-xpk-0.11.0.dist-info/top_level.txt,sha256=aDe4N0jicmuWExx_6w0TxWQJaEuPSs9BnLU-3aF1GLo,4
-xpk-0.11.0.dist-info/RECORD,,
+xpk-0.12.0.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+xpk-0.12.0.dist-info/METADATA,sha256=r_BW6P7rX9LHHQbFSLp_3WwEKfl2VFLYMFnyjMbSGes,71759
+xpk-0.12.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+xpk-0.12.0.dist-info/entry_points.txt,sha256=mzEtiIesFkT1kmcTUVDA1o3uOhiniX6tIz2wmOlMu1M,38
+xpk-0.12.0.dist-info/top_level.txt,sha256=aDe4N0jicmuWExx_6w0TxWQJaEuPSs9BnLU-3aF1GLo,4
+xpk-0.12.0.dist-info/RECORD,,