xpk 1.1.1__py3-none-any.whl → 1.1.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.
Files changed (8) hide show
  1. xpk/core/docker_container.py +3 -1
  2. xpk/core/docker_resources.py +5 -5
  3. {xpk-1.1.1.dist-info → xpk-1.1.2.dist-info}/METADATA +1 -1
  4. {xpk-1.1.1.dist-info → xpk-1.1.2.dist-info}/RECORD +8 -8
  5. {xpk-1.1.1.dist-info → xpk-1.1.2.dist-info}/WHEEL +1 -1
  6. {xpk-1.1.1.dist-info → xpk-1.1.2.dist-info}/entry_points.txt +0 -0
  7. {xpk-1.1.1.dist-info → xpk-1.1.2.dist-info}/licenses/LICENSE +0 -0
  8. {xpk-1.1.1.dist-info → xpk-1.1.2.dist-info}/top_level.txt +0 -0
xpk/core/docker_container.py CHANGED
@@ -181,7 +181,9 @@ def get_main_container(
181
181
  tpu_stacktrace_terminate_command=tpu_stacktrace_terminate_command,
182
182
  gpu_workload_terminate_command=gpu_workload_terminate_command,
183
183
  xpk_internal_commands=xpk_internal_commands,
184
- resources=get_main_container_resources(args, system, resource_type),
184
+ resources=get_main_container_resources(
185
+ args, system, resource_type, parallel_containers
186
+ ),
185
187
  volume_mounts=volume_mounts,
186
188
  )
187
189
  )
xpk/core/docker_resources.py CHANGED
@@ -23,7 +23,10 @@ from ..utils.execution_context import is_dry_run
23
23
 
24
24
 
25
25
  def get_main_container_resources(
26
- args, system: SystemCharacteristics, resource_type
26
+ args,
27
+ system: SystemCharacteristics,
28
+ resource_type: str,
29
+ parallel_containers: int,
27
30
  ) -> str:
28
31
  """Resources for the main container.
29
32
  Args:
@@ -53,10 +56,7 @@ def get_main_container_resources(
53
56
  offset_vCPUs = int(system.chips_per_vm) * 0.95
54
57
  return f'{resource_type}: {offset_vCPUs}'
55
58
 
56
- return (
57
- f'{resource_type}:'
58
- f' {int(system.chips_per_vm / system.parallel_containers)}'
59
- )
59
+ return f'{resource_type}: {int(system.chips_per_vm / parallel_containers)}'
60
60
 
61
61
 
62
62
  def get_env_container(args, system: SystemCharacteristics) -> str:
{xpk-1.1.1.dist-info → xpk-1.1.2.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: xpk
3
- Version: 1.1.1
3
+ Version: 1.1.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
{xpk-1.1.1.dist-info → xpk-1.1.2.dist-info}/RECORD RENAMED
@@ -38,10 +38,10 @@ xpk/core/cluster_test.py,sha256=VeC1C7kN0OJe6yeoL8GCaFk4uPhijP6CjvQAcE7q9xw,6653
38
38
  xpk/core/commands.py,sha256=at73VJHdZ4rVA8uvW997tNrvnCjP9v6zaw96bU0kd74,10841
39
39
  xpk/core/config.py,sha256=U2JDXx-XBuqQpZJf2iUDoww5--E8ejZfgmIxKeGu-gU,4668
40
40
  xpk/core/config_test.py,sha256=POSuofK0LFbNNygDAo2fjtKY4NMrRjUFeGcpBh9JOS4,3569
41
- xpk/core/docker_container.py,sha256=SZ3msIMx-2gBIok3j5N73KO1wVcMfzw1pQpgTdyZOQA,8243
41
+ xpk/core/docker_container.py,sha256=9kJpTEholW_d_GamjcqunCWT4XwrDyZs3fcvcPNCb8Y,8294
42
42
  xpk/core/docker_image.py,sha256=9vwqbb6Mc3C5ZEOph03WS-EWI5hxMYGGigqzIMkDTjE,6909
43
43
  xpk/core/docker_manager.py,sha256=vGPCWPDB507sxEsXvSD4IM-h5HqQzYLk7WSdCUmSDb4,10568
44
- xpk/core/docker_resources.py,sha256=7EXV1CvwCVogE5-m6utSE1GXxwf6EpB4QDYeuGXWHmI,12547
44
+ xpk/core/docker_resources.py,sha256=bwHGNh_gOtprVOeoFC8NObgKGD9aDjNc2XBMS6syD2Q,12562
45
45
  xpk/core/filestore.py,sha256=mcuUzsAPARbnrBG4fIGsEoN8NmzjaQ6k0tvIwMtjO9k,8068
46
46
  xpk/core/gcloud_context.py,sha256=d1wQ76zp7QMdG5BxB3sJz4b4OF5Mc8OzmPd_m0xd-Ys,6810
47
47
  xpk/core/gcloud_context_test.py,sha256=M8rp6S1zaEcAI7u4Bt8ukWKzv82HH5h9oYVojBcKgHk,5987
@@ -142,9 +142,9 @@ xpk/utils/validation.py,sha256=rE9LTkXJT7jIesodFb9pONL7ixhLqiQleyoaz7N39Dw,2765
142
142
  xpk/utils/validation_test.py,sha256=PEDSMUqZdt_Lx1FSR-LOTXKKtsJ47JH1fxugM0Gfz6Y,1168
143
143
  xpk/utils/versions.py,sha256=_Ep68W70a9605XjiaOOpBa9Is9jXlsoOiwL8v5Xt-WA,897
144
144
  xpk/utils/yaml.py,sha256=j8xuAJ9yAAwnQi6ozwZ-nMnDyDnc3xWkeBZMtSuP4RU,844
145
- xpk-1.1.1.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
146
- xpk-1.1.1.dist-info/METADATA,sha256=CqGGSACNDUVw7uPODTokMvMqpBMc1gDbY99AUmvQ68Q,10013
147
- xpk-1.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
148
- xpk-1.1.1.dist-info/entry_points.txt,sha256=mzEtiIesFkT1kmcTUVDA1o3uOhiniX6tIz2wmOlMu1M,38
149
- xpk-1.1.1.dist-info/top_level.txt,sha256=aDe4N0jicmuWExx_6w0TxWQJaEuPSs9BnLU-3aF1GLo,4
150
- xpk-1.1.1.dist-info/RECORD,,
145
+ xpk-1.1.2.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
146
+ xpk-1.1.2.dist-info/METADATA,sha256=qKqUDQuylrwPZI7NNzHvJWj6kJE08pZ3SbxHNGZ9qgI,10013
147
+ xpk-1.1.2.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
148
+ xpk-1.1.2.dist-info/entry_points.txt,sha256=mzEtiIesFkT1kmcTUVDA1o3uOhiniX6tIz2wmOlMu1M,38
149
+ xpk-1.1.2.dist-info/top_level.txt,sha256=aDe4N0jicmuWExx_6w0TxWQJaEuPSs9BnLU-3aF1GLo,4
150
+ xpk-1.1.2.dist-info/RECORD,,
{xpk-1.1.1.dist-info → xpk-1.1.2.dist-info}/WHEEL RENAMED
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (80.9.0)
2
+ Generator: setuptools (80.10.1)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5