xpk 0.14.2__py3-none-any.whl → 0.14.3__py3-none-any.whl

xpk/core/testing/commands_tester_test.py

@@ -0,0 +1,129 @@
+"""
+Copyright 2025 Google LLC
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+     https://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+import pytest
+from pytest_mock import MockerFixture
+
+from xpk.core.commands import run_command_for_value, run_command_with_updates_retry
+from xpk.core.testing.commands_tester import CommandsTester
+
+
+@pytest.fixture
+def mock_commands(mocker: MockerFixture) -> CommandsTester:
+  return CommandsTester(
+      mocker,
+      run_command_for_value_path=(
+          "xpk.core.testing.commands_tester_test.run_command_for_value"
+      ),
+      run_command_with_updates_retry_path=(
+          "xpk.core.testing.commands_tester_test.run_command_with_updates_retry"
+      ),
+  )
+
+
+def test_run_for_value_default_result(mock_commands: CommandsTester):
+  result = run_command_for_value(
+      command="cmd foo bar baz qux", task="Test command"
+  )
+
+  assert result == (0, "0")
+  mock_commands.assert_command_run("cmd", "bar")
+
+
+def test_run_command_with_updates_retry_default_result(
+    mock_commands: CommandsTester,
+):
+  result = run_command_with_updates_retry(
+      command="cmd foo bar baz qux", task="Test command"
+  )
+
+  assert result == 0
+  mock_commands.assert_command_run("cmd", "bar")
+
+
+def test_set_result_for_command(mock_commands: CommandsTester):
+  mock_commands.set_result_for_command((17, "Error!"), "cmd", "--err")
+
+  result = run_command_for_value("cmd foo --err", task="Matching test command")
+
+  assert result == (17, "Error!")
+
+
+def test_set_result_for_command_not_matching_other_commands(
+    mock_commands: CommandsTester,
+):
+  mock_commands.set_result_for_command((17, "Error!"), "cmd", "--err")
+
+  result = run_command_for_value(
+      "cmd foo bar", task="Not matching test command"
+  )
+
+  assert result == (0, "0")
+
+
+def test_assert_command_run(mock_commands: CommandsTester):
+  run_command_for_value("cmd foo bar", task="Test command")
+
+  mock_commands.assert_command_run("cmd foo bar")
+  mock_commands.assert_command_run("cmd")
+  mock_commands.assert_command_run("cmd", "bar")
+
+
+def test_assert_command_run_twice(mock_commands: CommandsTester):
+  run_command_for_value("cmd foo bar", task="Test command")
+  run_command_for_value("cmd foo bar", task="Test command")
+
+  mock_commands.assert_command_run("cmd", times=2)
+
+
+def test_assert_command_not_run(mock_commands: CommandsTester):
+  run_command_for_value("cmd", task="Test command")
+
+  mock_commands.assert_command_not_run("kubectl")
+
+
+def test_commands_history_contains_all_commands_in_order(
+    mock_commands: CommandsTester,
+):
+  run_command_for_value("cmd1", task="Test command")
+  run_command_for_value("cmd2 foo", task="Test command2")
+  run_command_for_value("cmd3 bar", task="Test command3")
+
+  assert mock_commands.commands_history == [
+      "cmd1",
+      "cmd2 foo",
+      "cmd3 bar",
+  ]
+
+
+def test_get_matching_commands(mock_commands: CommandsTester):
+  run_command_for_value("cmd", task="Test command")
+  run_command_for_value("cmd foo", task="Test command")
+  run_command_for_value("cmd foo bar", task="Test command")
+
+  assert len(mock_commands.get_matching_commands("cmd")) == 3
+  assert len(mock_commands.get_matching_commands("cmd", "foo")) == 2
+  assert mock_commands.get_matching_commands("cmd", "bar") == ["cmd foo bar"]
+
+
+def test_get_matching_commands_matches_parts_substrings(
+    mock_commands: CommandsTester,
+):
+  run_command_for_value("kubectl apply", task="Test command")
+  run_command_for_value("kubectl", task="Test command")
+
+  assert len(mock_commands.get_matching_commands("kube")) == 2
+  assert len(mock_commands.get_matching_commands("ctl apply")) == 1

xpk/core/updates.py

@@ -0,0 +1,57 @@
+"""
+Copyright 2025 Google LLC
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+     https://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+import json
+from json.decoder import JSONDecodeError
+from .commands import run_command_for_value
+from ..utils.console import xpk_print
+from ..utils.execution_context import is_dry_run
+from packaging.version import Version
+from .config import __version__
+
+
+def get_latest_xpk_version() -> tuple[int, Version | None]:
+  if is_dry_run():
+    return 0, Version(__version__)
+
+  return_code, result = run_command_for_value(
+      command="pip index versions xpk --json",
+      task="Retrieve latest XPK version",
+      quiet=True,
+  )
+
+  if return_code != 0:
+    return return_code, None
+
+  try:
+    parsed = json.loads(result.strip())
+    return 0, Version(parsed["latest"])
+  except JSONDecodeError:
+    return 1, None
+
+
+def print_xpk_hello() -> None:
+  current_version = Version(__version__)
+  xpk_print(f"Starting xpk v{current_version}", flush=True)
+  return_code, latest_version = get_latest_xpk_version()
+  if return_code != 0 or latest_version is None:
+    return
+  if current_version < latest_version:
+    xpk_print(
+        f"XPK version v{current_version} is outdated. Please consider upgrading"
+        f" to v{latest_version}",
+        flush=True,
+    )

xpk/core/updates_test.py

@@ -0,0 +1,80 @@
+"""
+Copyright 2025 Google LLC
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+     https://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+from ..utils.execution_context import set_dry_run
+from .updates import get_latest_xpk_version, print_xpk_hello
+from packaging.version import Version
+from .config import __version__
+from unittest.mock import MagicMock, patch
+
+
+def test_get_latest_xpk_version_returns_current_version_for_dry_run():
+  set_dry_run(True)
+  return_code, version = get_latest_xpk_version()
+  assert return_code == 0
+  assert version == Version(__version__)
+
+
+@patch('xpk.core.updates.run_command_for_value')
+def test_get_latest_xpk_version_returns_error_when_underlaying_command_errors(
+    run_command_for_value: MagicMock,
+):
+  run_command_for_value.return_value = (1, None)
+  set_dry_run(False)
+  return_code, version = get_latest_xpk_version()
+  assert return_code == 1
+  assert version is None
+
+
+@patch('xpk.core.updates.run_command_for_value')
+def test_get_latest_xpk_version_returns_version_returned_from_command(
+    run_command_for_value: MagicMock,
+):
+  run_command_for_value.return_value = (0, '{"latest": "1.0.0"}')
+  set_dry_run(False)
+  return_code, version = get_latest_xpk_version()
+  assert return_code == 0
+  assert version == Version('1.0.0')
+
+
+@patch('xpk.core.updates.xpk_print')
+@patch('xpk.core.updates.get_latest_xpk_version')
+def test_print_xpk_hello_does_not_print_update_when_version_check_fails(
+    get_latest_xpk_version: MagicMock, xpk_print: MagicMock
+):
+  get_latest_xpk_version.return_value = (1, None)
+  print_xpk_hello()
+  xpk_print.assert_called_once()
+
+
+@patch('xpk.core.updates.xpk_print')
+@patch('xpk.core.updates.get_latest_xpk_version')
+def test_print_xpk_hello_does_not_print_update_when_xpk_is_up_to_date(
+    get_latest_xpk_version: MagicMock, xpk_print: MagicMock
+):
+  get_latest_xpk_version.return_value = (0, Version(__version__))
+  print_xpk_hello()
+  xpk_print.assert_called_once()
+
+
+@patch('xpk.core.updates.xpk_print')
+@patch('xpk.core.updates.get_latest_xpk_version')
+def test_print_xpk_hello_prints_update_when_xpk_is_outdated(
+    get_latest_xpk_version: MagicMock, xpk_print: MagicMock
+):
+  get_latest_xpk_version.return_value = (0, Version('99.99.99'))
+  print_xpk_hello()
+  assert xpk_print.call_count == 2

xpk/main.py

@@ -35,8 +35,9 @@ import argparse
 import sys
 
 from .parser.core import set_parser
+from .core.updates import print_xpk_hello
 from .utils.console import xpk_print
-from .utils.execution_context import set_dry_run
+from .utils.execution_context import set_context
 ################### Compatibility Check ###################
 # Check that the user runs the below version or greater.
 
@@ -62,11 +63,13 @@ def main() -> None:
   parser = argparse.ArgumentParser(description='xpk command', prog='xpk')
   set_parser(parser=parser)
 
-  xpk_print('Starting xpk', flush=True)
   main_args = parser.parse_args()
   main_args.enable_ray_cluster = False
-  dry_run = 'dry_run' in main_args and main_args.dry_run
-  set_dry_run(dry_run)
+  set_context(
+      dry_run_value='dry_run' in main_args and main_args.dry_run,
+      quiet_value='quiet' in main_args and main_args.quiet,
+  )
+  print_xpk_hello()
   main_args.func(main_args)
   xpk_print('XPK Done.', flush=True)
 

xpk/parser/common.py

@@ -73,6 +73,14 @@ def add_shared_arguments(
       ),
       required=required,
   )
+  custom_parser_or_group.add_argument(
+      '--quiet',
+      type=bool,
+      action=argparse.BooleanOptionalAction,
+      default=False,
+      help='Disables prompting before unintended destructive actions.',
+      required=required,
+  )
 
 
 def add_cluster_arguments(

xpk/utils/execution_context.py

@@ -15,14 +15,32 @@ limitations under the License.
 """
 
 dry_run = False
+quiet = False
 
 
-def set_dry_run(value: bool) -> None:
+def set_context(dry_run_value: bool, quiet_value: bool) -> None:
+  """Sets the dry_run and quiet flags."""
+  set_dry_run(dry_run_value)
+  set_quiet(quiet_value)
+
+
+def set_dry_run(dry_run_value: bool) -> None:
   """Sets the dry_run flag."""
   global dry_run
-  dry_run = value
+  dry_run = dry_run_value
+
+
+def set_quiet(quiet_value: bool) -> None:
+  """Sets the quiet flag."""
+  global quiet
+  quiet = quiet_value
 
 
 def is_dry_run() -> bool:
   """Returns the current value of the dry_run flag."""
   return dry_run
+
+
+def is_quiet() -> bool:
+  """Returns the current value of the quiet flag."""
+  return quiet

{xpk-0.14.2.dist-info → xpk-0.14.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xpk
-Version: 0.14.2
+Version: 0.14.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
@@ -54,8 +54,6 @@ Dynamic: license-file
 
 [![Build Tests](https://github.com/google/xpk/actions/workflows/build_tests.yaml/badge.svg?query=branch%3Amain)](https://github.com/google/xpk/actions/workflows/build_tests.yaml?query=branch%3Amain)
 [![Nightly Tests](https://github.com/google/xpk/actions/workflows/nightly_tests.yaml/badge.svg?query=branch%3Amain)](https://github.com/google/xpk/actions/workflows/nightly_tests.yaml?query=branch%3Amain)
-[![Develop Tests](https://github.com/AI-Hypercomputer/xpk/actions/workflows/build_tests.yaml/badge.svg?query=branch%3Adevelop)](https://github.com/AI-Hypercomputer/xpk/actions/workflows/build_tests.yaml?query=branch%3Adevelop)
-[![Develop Nightly Tests](https://github.com/AI-Hypercomputer/xpk/actions/workflows/nightly_tests.yaml/badge.svg?query=branch%3Adevelop)](https://github.com/AI-Hypercomputer/xpk/actions/workflows/nightly_tests.yaml?query=branch%3Adevelop)
 
 # Overview
 

{xpk-0.14.2.dist-info → xpk-0.14.3.dist-info}/RECORD

@@ -5,73 +5,79 @@ integration/gcluster_a3ultra_test.py,sha256=8wEtlQN1_uIBUsidvH_l7Ab-ikDpnABrlu9k
 integration/gcluster_a4_test.py,sha256=GCe6BujHCvM62kIGOd-9Wvz-IrR0BY5d83bGD1cmsQ0,5754
 integration/gcluster_test.py,sha256=3GSOMszzNW6Yr4T4PFIpmszonwDAAGpSdKutUA77O-g,3304
 xpk/__init__.py,sha256=7mu-VQDQMyxM5To0KOhuYe4y2TYGsEkfV7hXZmUyih4,561
-xpk/main.py,sha256=7YBpzpHxV61c2Js0-uUHUXxTW7cWhLLdMc263rYRW3E,2416
+xpk/main.py,sha256=ZFZCouyUgghk9bE9lpRidwGGGutFBLPAazSt5g3oezY,2503
 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=Cj1bDpzPMoPdhaKKrOJJLJ3JzRvJrCMn8huQoHHIZJI,4192
-xpk/commands/cluster.py,sha256=CG4Ieg3QADNUAwyPzZh0P3mop4Pv1DILrNMJVLvgbi8,41682
+xpk/commands/cluster.py,sha256=YBWeqbF7b22iUoyr_jZEI2NUmADHRx7kB2wnMrT3ink,42828
 xpk/commands/cluster_gcluster.py,sha256=MOxQfQ19sxaDtDBfIzUdrxw4FpboDIiGFKEtFLwfEgQ,13080
-xpk/commands/cluster_gcluster_test.py,sha256=MoMlHbHnCI4ZrnMai1Zw71IP1ORxVVh23cMD5qgXOtQ,6136
-xpk/commands/cluster_test.py,sha256=tT1oEUXqNkzElgHZ-UXJGkJEGyakuaPb1cLHjV-eeM0,2800
+xpk/commands/cluster_gcluster_test.py,sha256=52pQprD7f3YWyO8Y0miULK35F8_G6tqKEUYBhMUv_Ug,6130
+xpk/commands/cluster_test.py,sha256=b6z7H0DuhQ7uP5eGBqLNX3BVqBBOWc3QYn5cKcolhyU,7357
 xpk/commands/common.py,sha256=nxLKPhXuAMW7wq-5xL2YeOgDTMUDcLGMujatOUHJ3s4,2504
 xpk/commands/config.py,sha256=gFNkf3ibsvZmcPpkpKXe-KJmHO5IKucNwLCXNgKvaDc,836
 xpk/commands/info.py,sha256=uhv5mPfgg9N-5JhQw4dT2jujL9ZC5kzGA18h9NFfm5A,7429
-xpk/commands/inspector.py,sha256=JruseZl9ZIlR9-Lv_pn8YHfLdyiMgHHSf7xPGBAtXTM,12616
+xpk/commands/inspector.py,sha256=7jA0indMDxBAR7iThPHTYKw-k6rQuZbcMTO5TGlp60w,12910
 xpk/commands/job.py,sha256=rPIfWvgm5mLz7K7YDLK721ZcUcg5OEmYVAPAtRtB5Ag,6718
-xpk/commands/kind.py,sha256=NM2CoKJXrYtY9DHZ2yM097WGcI1kno_G2b23tlLVIEI,7658
+xpk/commands/kind.py,sha256=7cqnKrt6gfjXN-coeh_tHBI6drvEOrcms3ZgGjMfbk0,7655
 xpk/commands/kjob_common.py,sha256=bRaORiGVjPAdN0T3aRmbcQgXYe-EtjoVKePdWzQ5xU4,1928
 xpk/commands/run.py,sha256=D0zgmnGeBLATphYhzQj29EScxrMmAKqPRhP6nfWuYcY,4085
 xpk/commands/shell.py,sha256=mRHMwm3Izzsue4bocekm82Rg_cPUaGMClSlvNzNXQ-o,4467
 xpk/commands/storage.py,sha256=90nz84_ut-uwdIVrxnLYq0K4uYpZWncKM-ZZ_2bRzcI,11505
 xpk/commands/version.py,sha256=k30rdLP9clUM8eeSwRFhpfzSb1qwcQImTfuC59Ed6CA,771
-xpk/commands/workload.py,sha256=S6g2B8bzhM_XNGY4NJUYVODcIuvutKQmdl0VV7jwGzw,29239
-xpk/commands/workload_test.py,sha256=e6D1j2Akewq50_76oD84gSCdcW44PEOsIxLPjaL11s8,2344
+xpk/commands/workload.py,sha256=LL2gO-pPhVgC94cA8bHL5HO8qpMB0jG6jLXR93HYwgk,30527
+xpk/commands/workload_test.py,sha256=WCBPem5C5-9HX_6SbLYbiiy6L7huecQKdDpjIG-qBxc,4755
 xpk/core/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
-xpk/core/capacity.py,sha256=ohrVvguNwSv7HufdhAhp9dEtf49QT9eXPaPuI9BuAck,7401
+xpk/core/capacity.py,sha256=vpUIXxwSVeyPQdyfxGBPRRU5Q9q_h07zqRhj-Mp22kg,7958
+xpk/core/capacity_test.py,sha256=rnIVNGJh3impyuk58ixwXfLn3z15GHhzNbLsWAn0i70,1514
 xpk/core/cluster.py,sha256=BoZzQO7j6Q5vfooFPCv72KDg7zYL_RFLLjxVEMKQsus,26359
 xpk/core/cluster_private.py,sha256=RLi0C7bV0NEUXl6QKQzvUT0weN9EdqPvjuuOQsNO0DY,6868
 xpk/core/commands.py,sha256=mtkT_ZsuHj5HTvFAj7mMd8kMXsWoWSA4zqHjlRiFL4o,10369
-xpk/core/config.py,sha256=EBGJC2MEmkks-9RpRsW7HFyfa4k-W-8geFIlkm3y44c,3407
+xpk/core/config.py,sha256=vB0HSi5TIPa4EPW5pAZCQ5SCbVrchtgzKZxCF4EY1Vk,3407
 xpk/core/config_test.py,sha256=v1qfyFRzLkYSQ7Wn4nx1N0dBSOFXidLWDfhkeHDZOVM,1847
-xpk/core/docker_container.py,sha256=GvkCJ2S5UKn8uh3pZhRd3X7iS0-PsQpRO8l7QhywVGc,7604
+xpk/core/docker_container.py,sha256=mibRro1MKX5eeV3p-7g7E9pKbMvDjg1e2Ma8p8oPwFw,7592
 xpk/core/docker_image.py,sha256=MIU397IGIPwkTZFK-ZGEWuc3RmUIF3sQQZUiUj2gLqA,6775
 xpk/core/docker_manager.py,sha256=JBFgyD6O7LKwEHJC7YuSoCDZqrFRtb-LjgWNqkfAbR0,10566
-xpk/core/docker_resources.py,sha256=_aKgpUjyJB2krQ1PkHrotB7K4kByLmPLbuvl_UVvuX8,12843
+xpk/core/docker_resources.py,sha256=bo_6fxvUOMY9XuL0Cps8mvn57K1osAVSgv0lpDGBGh8,12810
 xpk/core/filestore.py,sha256=mcuUzsAPARbnrBG4fIGsEoN8NmzjaQ6k0tvIwMtjO9k,8068
 xpk/core/gcloud_context.py,sha256=xZdVoRNLlE-kwXY5djoyQ0I0-KEh5nAohrVql7Jl42k,6649
 xpk/core/gcloud_context_test.py,sha256=YY0R6j-m62coVK2MAjWXDIdxdP6J5yn6R1RiTDkuExQ,2719
 xpk/core/gcluster_manager.py,sha256=lyv_MvdnkByy9_PEBj_ugAEBwnCbFNiWTSrEFjrMlPc,6236
 xpk/core/gcsfuse.py,sha256=kg5pgxdTjgiqquuGjev9fXzJPb8oiWPTK6wzCddzheQ,2125
 xpk/core/jobset.py,sha256=PJ4Fd8TNNLuYKNOMehoMYRIUEXyc5jsbHctJGqfW_8Y,4037
-xpk/core/kjob.py,sha256=symYO3DQfSNP6MAJE54QZuCPDF2kseaxZ-_mmsoBQjo,14478
-xpk/core/kueue_manager.py,sha256=PjZ1vjEfKvFdB5Tg5gK2qOm4J6LoHrpaafn5TizUo1A,13292
-xpk/core/kueue_manager_test.py,sha256=xN2viTlwrd5pPwn0OSn8RlD_gy13t3vpQfuN7jYFZWo,20062
+xpk/core/kjob.py,sha256=lDdsj6bKu6VC_SpAKUulKLY1hAGagCQeHwe-HqQwnX8,14457
+xpk/core/kueue_manager.py,sha256=39f0R-koMaMvx651aXlSJqpWvsoO26aafe0BcV-xQFg,13627
+xpk/core/kueue_manager_test.py,sha256=sSphrcfPLKVJ2cfvc5vPngrVWi5OBW9TX_T6ECSRg0w,13877
 xpk/core/monitoring.py,sha256=__bzTq_DIDAK8yIaN4F3MJh-yjYw5X1OlxmRgYOpf1g,4332
 xpk/core/mtc.py,sha256=pO7p3l-EzLFdTE8MdwWV8i0Zu-7epGql_kPoksVofIU,6259
-xpk/core/nap.py,sha256=uA33XccGjEF5RZRO5IpUMMzCf-u6D73cwwPEj4q1qvc,12820
+xpk/core/nap.py,sha256=BhuOPXmLhVs75ffnxtc5CaxF7yteEb1k_GwAKNc-olM,12817
 xpk/core/network.py,sha256=Oulb7U69lWkpOKxOC1C7ekJDpC51TLwd7XdZA3NQ7E0,10505
-xpk/core/nodepool.py,sha256=LfS_RvKHmS7f97hc_UidUC13FFEyGvfageSlt8d-5hw,23227
-xpk/core/nodepool_test.py,sha256=YNvp8WXznAI8DscrN9-BSnqII2AtRju2guxrobvCH8A,8805
-xpk/core/pathways.py,sha256=s-h_ofMrbFn3J6NFmT5OMe_HiUQIkI90ty7xbS05iA8,10710
+xpk/core/nodepool.py,sha256=2lTMtS8q9Q2CAu5XvgITufac2HMU7DLez84-CGzemDs,23237
+xpk/core/nodepool_test.py,sha256=q4LQEPkFWkozM8lrnbXUVJN1VbFjUdZsVZlFh2uVTmI,8793
+xpk/core/pathways.py,sha256=5b5WOpCmMYWKBz7SOxOXbxAPpAig7k06HSEqjXsOmeg,10707
 xpk/core/ray.py,sha256=JWhc_ToRHpF4_URGnuE_47FMgamaRsA4KVUMpqThWzw,6145
-xpk/core/resources.py,sha256=HlYNPQlaJa5y-pb70aVJzdiiOLDnoACF-wJXePM4ejs,8077
-xpk/core/scheduling.py,sha256=_o8QkQxVM-8z5K5ATslM_qQ87f5dzYE0ZeBcwQ6Oqic,9702
+xpk/core/resources.py,sha256=j1BAsIVvwnJ6zsPQDdkl_-SUY_MHumWcUqXPlXtGyLY,8074
+xpk/core/scheduling.py,sha256=MEcVxL-W9Is-3ufWGMwY9mCcQ4BkJsNJDq3oSWhh2L8,9645
 xpk/core/scheduling_test.py,sha256=m9KcglAbg0qly095PmrUOZxJYUE2UeQmkBNIWn5nFyk,979
 xpk/core/storage.py,sha256=NILvVAcLNMLmp4wKx_TEKbMMF5X1oL-FrQV46PT0_ds,16902
-xpk/core/system_characteristics.py,sha256=qHgqyQF6RVjU059YU0eB66_ia74tV-ogDyZbqCTUfw0,22589
-xpk/core/system_characteristics_test.py,sha256=MWvbaEQRoepYNZKOQcs73ppcTKW8plYQ9FQBotfA27I,2250
+xpk/core/system_characteristics.py,sha256=Ok9RUTpLnZh_O1UZpbOcztNEyLphRAlNlgRpGyQz0mo,22561
+xpk/core/system_characteristics_test.py,sha256=37RmnJuvu-Q6dBwiAOHBLsfvk48sML1K36BpWpdRk64,2303
+xpk/core/updates.py,sha256=sJCMyGh9epSz_96yeKMWGIvheqyMFEgxWFFQo7nzEBc,1723
+xpk/core/updates_test.py,sha256=oeGMv-wOHcOhWKGI4GnVLZ4Z_vCxLzLHeXRCi8rIDIM,2704
 xpk/core/vertex.py,sha256=orIZAVwZruRJQ6-vgc1wShuTsiipdH-zHQ9O4ie_HSA,3638
 xpk/core/workload.py,sha256=6TVZM15n8W7046VgmmH9Jv54MrhExtLQH3GaiwlV8Xs,8959
 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=VWhp89upY7vqVFUqcpWR6zwTBoFUSahX2sQxXEdUpSk,36393
+xpk/core/blueprint/blueprint_definitions.py,sha256=OgKkMnz4xAnmnZLaXuehVzNm1Gr2R4HWgofc7qd_mC4,2258
+xpk/core/blueprint/blueprint_generator.py,sha256=R4ReyN4syaq582wiGdcH6s0XCFr7UJ_P07U4IyvZJww,36407
 xpk/core/blueprint/blueprint_test.py,sha256=T058Dq-x4wQQqjs33BWhjHdT4qJLDwsIcc7vareHh_c,7204
 xpk/core/blueprint/testing/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
 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/testing/__init__.py,sha256=PkV8D9WOtlJHH5AIxsQaKeIBcmupT_Ol_bwJgN6G2I8,561
+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/storage_decorator.py,sha256=DDYQVO1OKTLhveDOA4V6b2RWr4n0fbwHdnoFFmW7iaQ,2000
@@ -82,7 +88,7 @@ xpk/parser/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
 xpk/parser/batch.py,sha256=mJU-Cp1yTLje59vD-B1IiBcUeD-ZmEsoeB4xhj9cflc,1406
 xpk/parser/cluster.py,sha256=1cc2pCMbceF6Lx6MXINlgEfiIRUYMITha3eCE6R8SrI,30708
 xpk/parser/cluster_test.py,sha256=h7lILF2V37WzQSZjGD2jCqI1iWJNY7BLODZf1uhK9QA,1806
-xpk/parser/common.py,sha256=1wihGYBe01ci4quZnkeR60-u_lapSOG4WhRFrE3yFR8,8108
+xpk/parser/common.py,sha256=zsCCzlaACO8xX0lpzquTRfHFmykOMjdHlsdo0YGmMq0,8348
 xpk/parser/config.py,sha256=-XnWx9aFsBW4Uzo_hpOMD2ZQ0bdZLvq1ksv83_5jqSM,1633
 xpk/parser/core.py,sha256=VRJerlS92ufoQbG1mZv7B04DAP4qGkBHa4pRXgcbAs0,4761
 xpk/parser/info.py,sha256=UJohxVVWdt9IgUXoPsrVae2DN1BjAVGWrSN2ajrB8RQ,1860
@@ -111,7 +117,7 @@ xpk/templates/storage.yaml,sha256=AykdyMtDnKZF8Y_0BYxoYP03hEIzEk6iNalXAQHgAls,16
 xpk/templates/volume_bundle.yaml,sha256=sqeag7GPWqGNQ5doZtO9IVAX_vKYRO73-aBE7waEtSY,129
 xpk/utils/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
 xpk/utils/console.py,sha256=hRbvtog_VAzuxt5GfwK5GZdd5SWaa7kvWG8zo_qFRQc,1519
-xpk/utils/execution_context.py,sha256=WYxm6NExBIP6iLAWaL5aV858riGJbAHn0Zs6fmKlmzE,784
+xpk/utils/execution_context.py,sha256=hONGz1hQSKE-puah2rE_uN9YUeEC4oW82VOryw5_Vgo,1181
 xpk/utils/feature_flags.py,sha256=b8zB_zEiKcoOnyGHc7vuQN0ruRTdY8ixqUjWT7Ilp-M,824
 xpk/utils/file.py,sha256=hi9v4gfwiB3JHi3tnelPbm_dlTUt47U0wvvWKQqMjiQ,2500
 xpk/utils/gcs_utils.py,sha256=zg-XSTv4G4TFjeT2bNBm2WLdDXPrOZi0rNv_JdppNg4,4113
@@ -125,9 +131,9 @@ xpk/utils/topology_test.py,sha256=jDXCPgBPfByqjhi0W9A5c8uOHOYjRav53nNlg71ipjk,19
 xpk/utils/validation.py,sha256=-Qd5jqkVzQHkJvmQnGjHjtAcfvz064Vbo6_sl4EnYKw,3497
 xpk/utils/validation_test.py,sha256=PEDSMUqZdt_Lx1FSR-LOTXKKtsJ47JH1fxugM0Gfz6Y,1168
 xpk/utils/yaml.py,sha256=j8xuAJ9yAAwnQi6ozwZ-nMnDyDnc3xWkeBZMtSuP4RU,844
-xpk-0.14.2.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-xpk-0.14.2.dist-info/METADATA,sha256=XP5pIZaZR0oSR9WDn_vbpq5zHRURGHGJ8kRuVw4-s0I,71934
-xpk-0.14.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-xpk-0.14.2.dist-info/entry_points.txt,sha256=mzEtiIesFkT1kmcTUVDA1o3uOhiniX6tIz2wmOlMu1M,38
-xpk-0.14.2.dist-info/top_level.txt,sha256=TQKZWgV7LSElvmunYT9V_627qOMoxq3qYzWAFzKudB8,16
-xpk-0.14.2.dist-info/RECORD,,
+xpk-0.14.3.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+xpk-0.14.3.dist-info/METADATA,sha256=HOH_X3b7o-ynhR0HK95rQrsO8VJm646oVmUpV9lHiRU,71468
+xpk-0.14.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+xpk-0.14.3.dist-info/entry_points.txt,sha256=mzEtiIesFkT1kmcTUVDA1o3uOhiniX6tIz2wmOlMu1M,38
+xpk-0.14.3.dist-info/top_level.txt,sha256=TQKZWgV7LSElvmunYT9V_627qOMoxq3qYzWAFzKudB8,16
+xpk-0.14.3.dist-info/RECORD,,