xpk 0.14.1__tar.gz → 0.14.2__tar.gz

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.
Files changed (138) hide show
  1. {xpk-0.14.1/src/xpk.egg-info → xpk-0.14.2}/PKG-INFO +1 -1
  2. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/cluster.py +1 -1
  3. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/config.py +1 -1
  4. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/kueue_manager.py +2 -2
  5. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/kueue_manager_test.py +1 -1
  6. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/templates/kueue_gke_default_topology.yaml.j2 +1 -1
  7. {xpk-0.14.1 → xpk-0.14.2/src/xpk.egg-info}/PKG-INFO +1 -1
  8. {xpk-0.14.1 → xpk-0.14.2}/LICENSE +0 -0
  9. {xpk-0.14.1 → xpk-0.14.2}/README.md +0 -0
  10. {xpk-0.14.1 → xpk-0.14.2}/pyproject.toml +0 -0
  11. {xpk-0.14.1 → xpk-0.14.2}/setup.cfg +0 -0
  12. {xpk-0.14.1 → xpk-0.14.2}/src/integration/__init__.py +0 -0
  13. {xpk-0.14.1 → xpk-0.14.2}/src/integration/docker_manager_test.py +0 -0
  14. {xpk-0.14.1 → xpk-0.14.2}/src/integration/gcluster_a3mega_test.py +0 -0
  15. {xpk-0.14.1 → xpk-0.14.2}/src/integration/gcluster_a3ultra_test.py +0 -0
  16. {xpk-0.14.1 → xpk-0.14.2}/src/integration/gcluster_a4_test.py +0 -0
  17. {xpk-0.14.1 → xpk-0.14.2}/src/integration/gcluster_test.py +0 -0
  18. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/__init__.py +0 -0
  19. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/api/__init__.py +0 -0
  20. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/api/storage_crd.yaml +0 -0
  21. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/__init__.py +0 -0
  22. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/batch.py +0 -0
  23. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/cluster.py +0 -0
  24. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/cluster_gcluster.py +0 -0
  25. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/cluster_gcluster_test.py +0 -0
  26. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/cluster_test.py +0 -0
  27. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/common.py +0 -0
  28. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/config.py +0 -0
  29. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/info.py +0 -0
  30. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/inspector.py +0 -0
  31. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/job.py +0 -0
  32. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/kind.py +0 -0
  33. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/kjob_common.py +0 -0
  34. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/run.py +0 -0
  35. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/shell.py +0 -0
  36. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/storage.py +0 -0
  37. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/version.py +0 -0
  38. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/workload.py +0 -0
  39. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/commands/workload_test.py +0 -0
  40. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/__init__.py +0 -0
  41. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/blueprint/__init__.py +0 -0
  42. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/blueprint/blueprint_definitions.py +0 -0
  43. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/blueprint/blueprint_generator.py +0 -0
  44. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/blueprint/blueprint_test.py +0 -0
  45. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/blueprint/testing/__init__.py +0 -0
  46. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/capacity.py +0 -0
  47. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/cluster_private.py +0 -0
  48. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/commands.py +0 -0
  49. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/config_test.py +0 -0
  50. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/docker_container.py +0 -0
  51. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/docker_image.py +0 -0
  52. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/docker_manager.py +0 -0
  53. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/docker_resources.py +0 -0
  54. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/filestore.py +0 -0
  55. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/gcloud_context.py +0 -0
  56. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/gcloud_context_test.py +0 -0
  57. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/gcluster_manager.py +0 -0
  58. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/gcsfuse.py +0 -0
  59. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/jobset.py +0 -0
  60. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/kjob.py +0 -0
  61. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/monitoring.py +0 -0
  62. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/mtc.py +0 -0
  63. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/nap.py +0 -0
  64. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/network.py +0 -0
  65. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/nodepool.py +0 -0
  66. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/nodepool_test.py +0 -0
  67. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/pathways.py +0 -0
  68. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/ray.py +0 -0
  69. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/remote_state/__init__.py +0 -0
  70. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/remote_state/fuse_remote_state.py +0 -0
  71. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/remote_state/remote_state_client.py +0 -0
  72. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/resources.py +0 -0
  73. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/scheduling.py +0 -0
  74. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/scheduling_test.py +0 -0
  75. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/storage.py +0 -0
  76. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/system_characteristics.py +0 -0
  77. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/system_characteristics_test.py +0 -0
  78. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/vertex.py +0 -0
  79. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/workload.py +0 -0
  80. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/workload_decorators/__init__.py +0 -0
  81. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/workload_decorators/rdma_decorator.py +0 -0
  82. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/workload_decorators/storage_decorator.py +0 -0
  83. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/workload_decorators/tcpx_decorator.py +0 -0
  84. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/workload_decorators/tcpx_decorator_test.py +0 -0
  85. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/workload_decorators/tcpxo_decorator.py +0 -0
  86. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/core/workload_test.py +0 -0
  87. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/main.py +0 -0
  88. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/parser/__init__.py +0 -0
  89. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/parser/batch.py +0 -0
  90. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/parser/cluster.py +0 -0
  91. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/parser/cluster_test.py +0 -0
  92. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/parser/common.py +0 -0
  93. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/parser/config.py +0 -0
  94. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/parser/core.py +0 -0
  95. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/parser/info.py +0 -0
  96. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/parser/inspector.py +0 -0
  97. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/parser/job.py +0 -0
  98. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/parser/kind.py +0 -0
  99. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/parser/run.py +0 -0
  100. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/parser/shell.py +0 -0
  101. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/parser/storage.py +0 -0
  102. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/parser/validators.py +0 -0
  103. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/parser/version.py +0 -0
  104. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/parser/workload.py +0 -0
  105. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/parser/workload_test.py +0 -0
  106. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/templates/__init__.py +0 -0
  107. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/templates/cluster_preheat.yaml.j2 +0 -0
  108. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/templates/filestore-pv.yaml +0 -0
  109. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/templates/filestore-pvc.yaml +0 -0
  110. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/templates/filestore-sc.yaml +0 -0
  111. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/templates/fuse-pv.yaml +0 -0
  112. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/templates/fuse-pvc.yaml +0 -0
  113. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/templates/kueue_config.yaml.j2 +0 -0
  114. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/templates/kueue_sub_slicing_topology.yaml.j2 +0 -0
  115. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/templates/mtc-cpc.yaml +0 -0
  116. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/templates/storage.yaml +0 -0
  117. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/templates/volume_bundle.yaml +0 -0
  118. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/utils/__init__.py +0 -0
  119. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/utils/console.py +0 -0
  120. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/utils/execution_context.py +0 -0
  121. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/utils/feature_flags.py +0 -0
  122. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/utils/file.py +0 -0
  123. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/utils/gcs_utils.py +0 -0
  124. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/utils/kubectl.py +0 -0
  125. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/utils/kueue.py +0 -0
  126. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/utils/network.py +0 -0
  127. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/utils/objects.py +0 -0
  128. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/utils/templates.py +0 -0
  129. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/utils/topology.py +0 -0
  130. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/utils/topology_test.py +0 -0
  131. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/utils/validation.py +0 -0
  132. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/utils/validation_test.py +0 -0
  133. {xpk-0.14.1 → xpk-0.14.2}/src/xpk/utils/yaml.py +0 -0
  134. {xpk-0.14.1 → xpk-0.14.2}/src/xpk.egg-info/SOURCES.txt +0 -0
  135. {xpk-0.14.1 → xpk-0.14.2}/src/xpk.egg-info/dependency_links.txt +0 -0
  136. {xpk-0.14.1 → xpk-0.14.2}/src/xpk.egg-info/entry_points.txt +0 -0
  137. {xpk-0.14.1 → xpk-0.14.2}/src/xpk.egg-info/requires.txt +0 -0
  138. {xpk-0.14.1 → xpk-0.14.2}/src/xpk.egg-info/top_level.txt +0 -0
@@ -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
@@ -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'
@@ -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
 
@@ -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
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes