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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
xpk/core/cluster.py CHANGED
@@ -37,7 +37,7 @@ from .resources import get_cluster_system_characteristics
37
37
  from .system_characteristics import SystemCharacteristics
38
38
 
39
39
  JOBSET_VERSION = 'v0.8.0'
40
- PATHWAYS_JOB_VERSION = 'v0.1.3'
40
+ PATHWAYS_JOB_VERSION = 'v0.1.4'
41
41
  INSTALLER_NCCL_TCPX = 'https://raw.githubusercontent.com/GoogleCloudPlatform/container-engine-accelerators/master/gpudirect-tcpx/nccl-tcpx-installer.yaml'
42
42
  INSTALLER_NCCL_TCPXO = 'https://raw.githubusercontent.com/GoogleCloudPlatform/container-engine-accelerators/master/gpudirect-tcpxo/nccl-tcpxo-installer.yaml'
43
43
  INSTALLER_NCCL_RDMA = 'https://raw.githubusercontent.com/GoogleCloudPlatform/container-engine-accelerators/master/gpudirect-rdma/nccl-rdma-installer.yaml'
xpk/core/config.py CHANGED
@@ -22,7 +22,7 @@ from ..utils import file
22
22
  from ..utils.console import xpk_print
23
23
 
24
24
  # This is the version for XPK PyPI package
25
- __version__ = 'v0.14.1'
25
+ __version__ = 'v0.14.2'
26
26
  XPK_CURRENT_VERSION = __version__
27
27
  XPK_CONFIG_FILE = os.path.expanduser('~/.config/xpk/config.yaml')
28
28
 
xpk/core/kueue_manager.py CHANGED
@@ -51,7 +51,7 @@ KUEUE_CONTROLLER_MANAGER_JINJA_FILE = "kueue_controller_manager.yaml.j2"
51
51
  KUEUE_SUB_SLICING_TOPOLOGY_JINJA_FILE = "kueue_sub_slicing_topology.yaml.j2"
52
52
  MEMORY_SIZE_PER_VM = 1.2
53
53
  MIN_MEMORY_LIMIT_SIZE = 4096
54
- KUEUE_VERSION = "v0.14.1"
54
+ KUEUE_VERSION = "v0.12.2"
55
55
 
56
56
 
57
57
  @dataclass
@@ -131,7 +131,7 @@ class KueueManager:
131
131
  command,
132
132
  task,
133
133
  dry_run_return_val="""
134
- v0.14.1""",
134
+ v0.12.1""",
135
135
  )
136
136
  if return_code != 0:
137
137
  return return_code, None
@@ -83,7 +83,7 @@ class KueueManagerTest(unittest.TestCase):
83
83
  self, mock_configure, mock_install, mock_get_version
84
84
  ):
85
85
  """Test install_or_upgrade when Kueue is already up to date."""
86
- mock_get_version.return_value = (0, "v0.99.9")
86
+ mock_get_version.return_value = (0, "v0.12.3")
87
87
  kueue_config = MagicMock(spec=KueueConfig)
88
88
 
89
89
  result = self.kueue_manager.install_or_upgrade(kueue_config)
@@ -1,4 +1,4 @@
1
- apiVersion: kueue.x-k8s.io/v1beta1
1
+ apiVersion: kueue.x-k8s.io/v1alpha1
2
2
  kind: Topology
3
3
  metadata:
4
4
  name: "gke-default"
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: xpk
3
- Version: 0.14.1
3
+ Version: 0.14.2
4
4
  Summary: xpk helps Cloud developers to orchestrate training jobs on accelerators on GKE.
5
5
  Author-email: XPK team <xpk-code-reviewers@google.com>
6
6
  License: Apache-2.0
@@ -29,10 +29,10 @@ xpk/commands/workload.py,sha256=S6g2B8bzhM_XNGY4NJUYVODcIuvutKQmdl0VV7jwGzw,2923
29
29
  xpk/commands/workload_test.py,sha256=e6D1j2Akewq50_76oD84gSCdcW44PEOsIxLPjaL11s8,2344
30
30
  xpk/core/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
31
31
  xpk/core/capacity.py,sha256=ohrVvguNwSv7HufdhAhp9dEtf49QT9eXPaPuI9BuAck,7401
32
- xpk/core/cluster.py,sha256=H5-x0xYijQbgINbAIXovu6JoR-9iKnuVQrMVy1xrEP0,26359
32
+ xpk/core/cluster.py,sha256=BoZzQO7j6Q5vfooFPCv72KDg7zYL_RFLLjxVEMKQsus,26359
33
33
  xpk/core/cluster_private.py,sha256=RLi0C7bV0NEUXl6QKQzvUT0weN9EdqPvjuuOQsNO0DY,6868
34
34
  xpk/core/commands.py,sha256=mtkT_ZsuHj5HTvFAj7mMd8kMXsWoWSA4zqHjlRiFL4o,10369
35
- xpk/core/config.py,sha256=G7MKXARftcuoKSV90F40HXh7kYj1ePde8O8dCp95QoI,3407
35
+ xpk/core/config.py,sha256=EBGJC2MEmkks-9RpRsW7HFyfa4k-W-8geFIlkm3y44c,3407
36
36
  xpk/core/config_test.py,sha256=v1qfyFRzLkYSQ7Wn4nx1N0dBSOFXidLWDfhkeHDZOVM,1847
37
37
  xpk/core/docker_container.py,sha256=GvkCJ2S5UKn8uh3pZhRd3X7iS0-PsQpRO8l7QhywVGc,7604
38
38
  xpk/core/docker_image.py,sha256=MIU397IGIPwkTZFK-ZGEWuc3RmUIF3sQQZUiUj2gLqA,6775
@@ -45,8 +45,8 @@ xpk/core/gcluster_manager.py,sha256=lyv_MvdnkByy9_PEBj_ugAEBwnCbFNiWTSrEFjrMlPc,
45
45
  xpk/core/gcsfuse.py,sha256=kg5pgxdTjgiqquuGjev9fXzJPb8oiWPTK6wzCddzheQ,2125
46
46
  xpk/core/jobset.py,sha256=PJ4Fd8TNNLuYKNOMehoMYRIUEXyc5jsbHctJGqfW_8Y,4037
47
47
  xpk/core/kjob.py,sha256=symYO3DQfSNP6MAJE54QZuCPDF2kseaxZ-_mmsoBQjo,14478
48
- xpk/core/kueue_manager.py,sha256=TFdl33N-b-eSrOrsnJueP1H4qYWpkX_lFDhLSVvfeGM,13292
49
- xpk/core/kueue_manager_test.py,sha256=PtsSSfwtdk-2wK_dr_OQb8oYWVce5nmDN6iowND_3b4,20062
48
+ xpk/core/kueue_manager.py,sha256=PjZ1vjEfKvFdB5Tg5gK2qOm4J6LoHrpaafn5TizUo1A,13292
49
+ xpk/core/kueue_manager_test.py,sha256=xN2viTlwrd5pPwn0OSn8RlD_gy13t3vpQfuN7jYFZWo,20062
50
50
  xpk/core/monitoring.py,sha256=__bzTq_DIDAK8yIaN4F3MJh-yjYw5X1OlxmRgYOpf1g,4332
51
51
  xpk/core/mtc.py,sha256=pO7p3l-EzLFdTE8MdwWV8i0Zu-7epGql_kPoksVofIU,6259
52
52
  xpk/core/nap.py,sha256=uA33XccGjEF5RZRO5IpUMMzCf-u6D73cwwPEj4q1qvc,12820
@@ -104,7 +104,7 @@ xpk/templates/filestore-sc.yaml,sha256=vHzcU7jk0B5z7EgTfQmMM1m2TIzR70ny6MvthyYdq
104
104
  xpk/templates/fuse-pv.yaml,sha256=-CM6AYAy4HKOErd9ogiM-6vWIuWS5yXoXyBAp2EoZsM,321
105
105
  xpk/templates/fuse-pvc.yaml,sha256=heGWvRIetukAI9pH9auXxnU2H-G_8pL_wRIzoWDLVH8,218
106
106
  xpk/templates/kueue_config.yaml.j2,sha256=ZAZwzZ28piwlXi9Offo2CvrQ3K9gv1r9BJTPavnUEdY,2055
107
- xpk/templates/kueue_gke_default_topology.yaml.j2,sha256=wW3qt6p3VDPgFVX7Ozw4-O4QgQ3mhH8U3osKnmuOFaE,299
107
+ xpk/templates/kueue_gke_default_topology.yaml.j2,sha256=GDFUf0DUnVTVjjSVRqfyEhQoDj2sEoL2YWhHYJZU9N4,300
108
108
  xpk/templates/kueue_sub_slicing_topology.yaml.j2,sha256=Lv54uS3fC7sJNJdk4DFtnOiLSgq4wq38n_BQCx4Tz6Q,542
109
109
  xpk/templates/mtc-cpc.yaml,sha256=MPx75tog09kjRAvHoNOPCEobigQ17d7pYCUnZCevSDQ,340
110
110
  xpk/templates/storage.yaml,sha256=AykdyMtDnKZF8Y_0BYxoYP03hEIzEk6iNalXAQHgAls,163
@@ -125,9 +125,9 @@ xpk/utils/topology_test.py,sha256=jDXCPgBPfByqjhi0W9A5c8uOHOYjRav53nNlg71ipjk,19
125
125
  xpk/utils/validation.py,sha256=-Qd5jqkVzQHkJvmQnGjHjtAcfvz064Vbo6_sl4EnYKw,3497
126
126
  xpk/utils/validation_test.py,sha256=PEDSMUqZdt_Lx1FSR-LOTXKKtsJ47JH1fxugM0Gfz6Y,1168
127
127
  xpk/utils/yaml.py,sha256=j8xuAJ9yAAwnQi6ozwZ-nMnDyDnc3xWkeBZMtSuP4RU,844
128
- xpk-0.14.1.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
129
- xpk-0.14.1.dist-info/METADATA,sha256=41WVkaR-Bgj64aC9vi26hgI9g6CpnDtEF1G3yXhbpzI,71934
130
- xpk-0.14.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
131
- xpk-0.14.1.dist-info/entry_points.txt,sha256=mzEtiIesFkT1kmcTUVDA1o3uOhiniX6tIz2wmOlMu1M,38
132
- xpk-0.14.1.dist-info/top_level.txt,sha256=TQKZWgV7LSElvmunYT9V_627qOMoxq3qYzWAFzKudB8,16
133
- xpk-0.14.1.dist-info/RECORD,,
128
+ xpk-0.14.2.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
129
+ xpk-0.14.2.dist-info/METADATA,sha256=XP5pIZaZR0oSR9WDn_vbpq5zHRURGHGJ8kRuVw4-s0I,71934
130
+ xpk-0.14.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
131
+ xpk-0.14.2.dist-info/entry_points.txt,sha256=mzEtiIesFkT1kmcTUVDA1o3uOhiniX6tIz2wmOlMu1M,38
132
+ xpk-0.14.2.dist-info/top_level.txt,sha256=TQKZWgV7LSElvmunYT9V_627qOMoxq3qYzWAFzKudB8,16
133
+ xpk-0.14.2.dist-info/RECORD,,
File without changes