xpk 0.17.3__py3-none-any.whl → 1.0.0__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.
@@ -142,6 +142,7 @@ class SystemCharacteristics:
142
142
  device_type: str
143
143
  supports_sub_slicing: bool
144
144
  supports_super_slicing: bool
145
+ supports_accelerator_network_profile: bool
145
146
  docker_platform: DockerPlatform
146
147
  requires_workload_policy: bool = False
147
148
  gpu_config: Optional[GpuConfig] = None
@@ -233,6 +234,7 @@ def get_tpu_system_characteristics_map(
233
234
  machine_type: str,
234
235
  supported_topologies: list[str],
235
236
  docker_platform: DockerPlatform,
237
+ supports_accelerator_network_profile: bool,
236
238
  tpu_type_requires_workload_policy: bool = False,
237
239
  default_topologies: set[str] | None = None,
238
240
  sub_slicing_topologies: set[str] | None = None,
@@ -259,6 +261,7 @@ def get_tpu_system_characteristics_map(
259
261
  and vms_per_slice > 1,
260
262
  supports_sub_slicing=topology in sub_slicing_topologies,
261
263
  supports_super_slicing=topology in super_slicing_topologies,
264
+ supports_accelerator_network_profile=supports_accelerator_network_profile,
262
265
  docker_platform=docker_platform,
263
266
  )
264
267
  system_characteristics_map[f'{prefix}-{topology}'] = system
@@ -303,6 +306,7 @@ UserFacingNameToSystemCharacteristics = {
303
306
  device_type='l4-1',
304
307
  supports_sub_slicing=False,
305
308
  supports_super_slicing=False,
309
+ supports_accelerator_network_profile=False,
306
310
  gpu_config=GpuConfig(requires_topology=False),
307
311
  docker_platform=AMD_PLATFORM,
308
312
  ),
@@ -316,6 +320,7 @@ UserFacingNameToSystemCharacteristics = {
316
320
  device_type='l4-2',
317
321
  supports_sub_slicing=False,
318
322
  supports_super_slicing=False,
323
+ supports_accelerator_network_profile=False,
319
324
  gpu_config=GpuConfig(requires_topology=False),
320
325
  docker_platform=AMD_PLATFORM,
321
326
  ),
@@ -329,6 +334,7 @@ UserFacingNameToSystemCharacteristics = {
329
334
  device_type='l4-4',
330
335
  supports_sub_slicing=False,
331
336
  supports_super_slicing=False,
337
+ supports_accelerator_network_profile=False,
332
338
  gpu_config=GpuConfig(requires_topology=False),
333
339
  docker_platform=AMD_PLATFORM,
334
340
  ),
@@ -342,6 +348,7 @@ UserFacingNameToSystemCharacteristics = {
342
348
  device_type='l4-8',
343
349
  supports_sub_slicing=False,
344
350
  supports_super_slicing=False,
351
+ supports_accelerator_network_profile=False,
345
352
  gpu_config=GpuConfig(requires_topology=False),
346
353
  docker_platform=AMD_PLATFORM,
347
354
  ),
@@ -356,6 +363,7 @@ UserFacingNameToSystemCharacteristics = {
356
363
  device_type='a100-40gb-1',
357
364
  supports_sub_slicing=False,
358
365
  supports_super_slicing=False,
366
+ supports_accelerator_network_profile=False,
359
367
  gpu_config=GpuConfig(requires_topology=False),
360
368
  docker_platform=AMD_PLATFORM,
361
369
  ),
@@ -369,6 +377,7 @@ UserFacingNameToSystemCharacteristics = {
369
377
  device_type='a100-40gb-2',
370
378
  supports_sub_slicing=False,
371
379
  supports_super_slicing=False,
380
+ supports_accelerator_network_profile=False,
372
381
  gpu_config=GpuConfig(requires_topology=False),
373
382
  docker_platform=AMD_PLATFORM,
374
383
  ),
@@ -382,6 +391,7 @@ UserFacingNameToSystemCharacteristics = {
382
391
  device_type='a100-40gb-4',
383
392
  supports_sub_slicing=False,
384
393
  supports_super_slicing=False,
394
+ supports_accelerator_network_profile=False,
385
395
  gpu_config=GpuConfig(requires_topology=False),
386
396
  docker_platform=AMD_PLATFORM,
387
397
  ),
@@ -395,6 +405,7 @@ UserFacingNameToSystemCharacteristics = {
395
405
  device_type='a100-40gb-8',
396
406
  supports_sub_slicing=False,
397
407
  supports_super_slicing=False,
408
+ supports_accelerator_network_profile=False,
398
409
  gpu_config=GpuConfig(requires_topology=False),
399
410
  docker_platform=AMD_PLATFORM,
400
411
  ),
@@ -408,6 +419,7 @@ UserFacingNameToSystemCharacteristics = {
408
419
  device_type='gb200-4',
409
420
  supports_sub_slicing=False,
410
421
  supports_super_slicing=False,
422
+ supports_accelerator_network_profile=True,
411
423
  gpu_config=GpuConfig(
412
424
  requires_topology=True,
413
425
  nccl_installer=INSTALLER_NCCL_RDMA_A4X,
@@ -426,6 +438,7 @@ UserFacingNameToSystemCharacteristics = {
426
438
  device_type='gb200-4',
427
439
  supports_sub_slicing=False,
428
440
  supports_super_slicing=False,
441
+ supports_accelerator_network_profile=True,
429
442
  gpu_config=GpuConfig(
430
443
  requires_topology=True,
431
444
  nccl_installer=INSTALLER_NCCL_RDMA_A4X,
@@ -444,6 +457,7 @@ UserFacingNameToSystemCharacteristics = {
444
457
  device_type='b200-8',
445
458
  supports_sub_slicing=False,
446
459
  supports_super_slicing=False,
460
+ supports_accelerator_network_profile=True,
447
461
  gpu_config=GpuConfig(
448
462
  requires_topology=True,
449
463
  nccl_installer=INSTALLER_NCCL_RDMA,
@@ -462,6 +476,7 @@ UserFacingNameToSystemCharacteristics = {
462
476
  device_type='h200-141gb-8',
463
477
  supports_sub_slicing=False,
464
478
  supports_super_slicing=False,
479
+ supports_accelerator_network_profile=True,
465
480
  gpu_config=GpuConfig(
466
481
  requires_topology=True,
467
482
  nccl_installer=INSTALLER_NCCL_RDMA,
@@ -481,6 +496,7 @@ UserFacingNameToSystemCharacteristics = {
481
496
  device_type='h100-80gb-8',
482
497
  supports_sub_slicing=False,
483
498
  supports_super_slicing=False,
499
+ supports_accelerator_network_profile=True,
484
500
  gpu_config=GpuConfig(
485
501
  requires_topology=True,
486
502
  nccl_installer=INSTALLER_NCCL_TCPX,
@@ -500,6 +516,7 @@ UserFacingNameToSystemCharacteristics = {
500
516
  device_type='h100-mega-80gb-8',
501
517
  supports_sub_slicing=False,
502
518
  supports_super_slicing=False,
519
+ supports_accelerator_network_profile=True,
503
520
  gpu_config=GpuConfig(
504
521
  requires_topology=True,
505
522
  nccl_installer=INSTALLER_NCCL_TCPXO,
@@ -516,6 +533,7 @@ UserFacingNameToSystemCharacteristics = {
516
533
  machine_type='tpu7x-standard-1t',
517
534
  supported_topologies=['1x1x1'],
518
535
  tpu_type_requires_workload_policy=True,
536
+ supports_accelerator_network_profile=False,
519
537
  docker_platform=AMD_PLATFORM,
520
538
  ),
521
539
  **get_tpu_system_characteristics_map(
@@ -524,6 +542,7 @@ UserFacingNameToSystemCharacteristics = {
524
542
  gke_accelerator='tpu7x',
525
543
  machine_type='tpu7x-standard-4t',
526
544
  tpu_type_requires_workload_policy=True,
545
+ supports_accelerator_network_profile=False,
527
546
  docker_platform=AMD_PLATFORM,
528
547
  supported_topologies=generate_tpu_topologies(max_cubes=144),
529
548
  super_slicing_topologies=set(['4x4x4']),
@@ -635,6 +654,7 @@ UserFacingNameToSystemCharacteristics = {
635
654
  machine_type='ct6e-standard-1t',
636
655
  supported_topologies=['1x1'],
637
656
  docker_platform=AMD_PLATFORM,
657
+ supports_accelerator_network_profile=True,
638
658
  ),
639
659
  **get_tpu_system_characteristics_map(
640
660
  prefix='v6e',
@@ -644,6 +664,7 @@ UserFacingNameToSystemCharacteristics = {
644
664
  supported_topologies=['2x2'] + SUB_SLICING_TOPOLOGIES,
645
665
  sub_slicing_topologies=set(SUB_SLICING_TOPOLOGIES),
646
666
  docker_platform=AMD_PLATFORM,
667
+ supports_accelerator_network_profile=True,
647
668
  ),
648
669
  **get_tpu_system_characteristics_map(
649
670
  prefix='v5p',
@@ -652,6 +673,7 @@ UserFacingNameToSystemCharacteristics = {
652
673
  machine_type='ct5p-hightpu-4t',
653
674
  docker_platform=AMD_PLATFORM,
654
675
  supported_topologies=generate_tpu_topologies(max_cubes=140),
676
+ supports_accelerator_network_profile=False,
655
677
  default_topologies=set([
656
678
  '2x2x1',
657
679
  '2x2x2',
@@ -758,6 +780,7 @@ UserFacingNameToSystemCharacteristics = {
758
780
  machine_type='ct5lp-hightpu-4t',
759
781
  docker_platform=AMD_PLATFORM,
760
782
  supported_topologies=['2x4', '4x4', '4x8', '8x8', '8x16', '16x16'],
783
+ supports_accelerator_network_profile=False,
761
784
  ),
762
785
  **get_tpu_system_characteristics_map(
763
786
  prefix='v4',
@@ -768,6 +791,7 @@ UserFacingNameToSystemCharacteristics = {
768
791
  supported_topologies=generate_tpu_topologies(
769
792
  max_cubes=64, enforce_nondecreasing=False
770
793
  ),
794
+ supports_accelerator_network_profile=False,
771
795
  default_topologies=set([
772
796
  '2x2x1',
773
797
  '2x2x2',
@@ -796,6 +820,7 @@ UserFacingNameToSystemCharacteristics = {
796
820
  device_type='m1-megamem-96-1',
797
821
  supports_sub_slicing=False,
798
822
  supports_super_slicing=False,
823
+ supports_accelerator_network_profile=False,
799
824
  docker_platform=AMD_PLATFORM,
800
825
  ),
801
826
  # n2-standard-#vCPUs-#VMs
@@ -809,6 +834,7 @@ UserFacingNameToSystemCharacteristics = {
809
834
  device_type='n2-standard-64-1',
810
835
  supports_sub_slicing=False,
811
836
  supports_super_slicing=False,
837
+ supports_accelerator_network_profile=False,
812
838
  docker_platform=AMD_PLATFORM,
813
839
  ),
814
840
  'n2-standard-32-1': SystemCharacteristics(
@@ -821,6 +847,7 @@ UserFacingNameToSystemCharacteristics = {
821
847
  device_type='n2-standard-32-1',
822
848
  supports_sub_slicing=False,
823
849
  supports_super_slicing=False,
850
+ supports_accelerator_network_profile=False,
824
851
  docker_platform=AMD_PLATFORM,
825
852
  ),
826
853
  'n2-standard-32-2': SystemCharacteristics(
@@ -833,6 +860,7 @@ UserFacingNameToSystemCharacteristics = {
833
860
  device_type='n2-standard-32-2',
834
861
  supports_sub_slicing=False,
835
862
  supports_super_slicing=False,
863
+ supports_accelerator_network_profile=False,
836
864
  docker_platform=AMD_PLATFORM,
837
865
  ),
838
866
  'n2-standard-32-4': SystemCharacteristics(
@@ -845,6 +873,7 @@ UserFacingNameToSystemCharacteristics = {
845
873
  device_type='n2-standard-32-4',
846
874
  supports_sub_slicing=False,
847
875
  supports_super_slicing=False,
876
+ supports_accelerator_network_profile=False,
848
877
  docker_platform=AMD_PLATFORM,
849
878
  ),
850
879
  'n2-standard-32-8': SystemCharacteristics(
@@ -857,6 +886,7 @@ UserFacingNameToSystemCharacteristics = {
857
886
  device_type='n2-standard-32-8',
858
887
  supports_sub_slicing=False,
859
888
  supports_super_slicing=False,
889
+ supports_accelerator_network_profile=False,
860
890
  docker_platform=AMD_PLATFORM,
861
891
  ),
862
892
  'n2-standard-32-16': SystemCharacteristics(
@@ -869,6 +899,7 @@ UserFacingNameToSystemCharacteristics = {
869
899
  device_type='n2-standard-32-16',
870
900
  supports_sub_slicing=False,
871
901
  supports_super_slicing=False,
902
+ supports_accelerator_network_profile=False,
872
903
  docker_platform=AMD_PLATFORM,
873
904
  ),
874
905
  'n2-standard-32-32': SystemCharacteristics(
@@ -881,6 +912,7 @@ UserFacingNameToSystemCharacteristics = {
881
912
  device_type='n2-standard-32-32',
882
913
  supports_sub_slicing=False,
883
914
  supports_super_slicing=False,
915
+ supports_accelerator_network_profile=False,
884
916
  docker_platform=AMD_PLATFORM,
885
917
  ),
886
918
  'n2-standard-32-64': SystemCharacteristics(
@@ -893,6 +925,7 @@ UserFacingNameToSystemCharacteristics = {
893
925
  device_type='n2-standard-32-64',
894
926
  supports_sub_slicing=False,
895
927
  supports_super_slicing=False,
928
+ supports_accelerator_network_profile=False,
896
929
  docker_platform=AMD_PLATFORM,
897
930
  ),
898
931
  'n2-standard-32-128': SystemCharacteristics(
@@ -905,6 +938,7 @@ UserFacingNameToSystemCharacteristics = {
905
938
  device_type='n2-standard-32-128',
906
939
  supports_sub_slicing=False,
907
940
  supports_super_slicing=False,
941
+ supports_accelerator_network_profile=False,
908
942
  docker_platform=AMD_PLATFORM,
909
943
  ),
910
944
  'n2-standard-32-256': SystemCharacteristics(
@@ -917,6 +951,7 @@ UserFacingNameToSystemCharacteristics = {
917
951
  device_type='n2-standard-32-256',
918
952
  supports_sub_slicing=False,
919
953
  supports_super_slicing=False,
954
+ supports_accelerator_network_profile=False,
920
955
  docker_platform=AMD_PLATFORM,
921
956
  ),
922
957
  'n2-standard-32-512': SystemCharacteristics(
@@ -929,6 +964,7 @@ UserFacingNameToSystemCharacteristics = {
929
964
  device_type='n2-standard-32-512',
930
965
  supports_sub_slicing=False,
931
966
  supports_super_slicing=False,
967
+ supports_accelerator_network_profile=False,
932
968
  docker_platform=AMD_PLATFORM,
933
969
  ),
934
970
  'n2-standard-32-1024': SystemCharacteristics(
@@ -941,6 +977,7 @@ UserFacingNameToSystemCharacteristics = {
941
977
  device_type='n2-standard-32-1024',
942
978
  supports_sub_slicing=False,
943
979
  supports_super_slicing=False,
980
+ supports_accelerator_network_profile=False,
944
981
  docker_platform=AMD_PLATFORM,
945
982
  ),
946
983
  'n2-standard-32-2048': SystemCharacteristics(
@@ -953,6 +990,7 @@ UserFacingNameToSystemCharacteristics = {
953
990
  device_type='n2-standard-32-2048',
954
991
  supports_sub_slicing=False,
955
992
  supports_super_slicing=False,
993
+ supports_accelerator_network_profile=False,
956
994
  docker_platform=AMD_PLATFORM,
957
995
  ),
958
996
  }
@@ -34,6 +34,7 @@ def test_get_tpu_system_characteristics_map_returns_correct_values_for_1x1_topol
34
34
  supported_topologies=["1x1"],
35
35
  docker_platform=DockerPlatform.AMD,
36
36
  tpu_type_requires_workload_policy=False,
37
+ supports_accelerator_network_profile=False,
37
38
  )
38
39
 
39
40
  expected_system_characteristics = SystemCharacteristics(
@@ -48,6 +49,7 @@ def test_get_tpu_system_characteristics_map_returns_correct_values_for_1x1_topol
48
49
  supports_super_slicing=False,
49
50
  docker_platform=DockerPlatform.AMD,
50
51
  requires_workload_policy=False,
52
+ supports_accelerator_network_profile=False,
51
53
  )
52
54
  assert result == {
53
55
  "test-1": expected_system_characteristics,
@@ -62,6 +64,7 @@ def test_get_tpu_system_characteristics_map_returns_correct_values_for_2x2_topol
62
64
  gke_accelerator="test",
63
65
  machine_type="test",
64
66
  supported_topologies=["2x2"],
67
+ supports_accelerator_network_profile=False,
65
68
  docker_platform=DockerPlatform.AMD,
66
69
  tpu_type_requires_workload_policy=True,
67
70
  )
@@ -76,6 +79,7 @@ def test_get_tpu_system_characteristics_map_returns_correct_values_for_2x2_topol
76
79
  device_type="test-8",
77
80
  supports_sub_slicing=False,
78
81
  supports_super_slicing=False,
82
+ supports_accelerator_network_profile=False,
79
83
  docker_platform=DockerPlatform.AMD,
80
84
  requires_workload_policy=False,
81
85
  )
@@ -94,6 +98,7 @@ def test_get_tpu_system_characteristics_map_returns_correct_values_for_2x2x2_top
94
98
  supported_topologies=["2x2x2"],
95
99
  docker_platform=DockerPlatform.AMD,
96
100
  tpu_type_requires_workload_policy=True,
101
+ supports_accelerator_network_profile=False,
97
102
  )
98
103
 
99
104
  expected_system_characteristics = SystemCharacteristics(
@@ -106,6 +111,7 @@ def test_get_tpu_system_characteristics_map_returns_correct_values_for_2x2x2_top
106
111
  device_type="test-16",
107
112
  supports_sub_slicing=False,
108
113
  supports_super_slicing=False,
114
+ supports_accelerator_network_profile=False,
109
115
  docker_platform=DockerPlatform.AMD,
110
116
  requires_workload_policy=True,
111
117
  )
@@ -122,6 +128,7 @@ def test_get_tpu_system_characteristics_map_sets_sub_slicing_support():
122
128
  gke_accelerator="test",
123
129
  machine_type="test",
124
130
  supported_topologies=["4x4x4", "4x4x8", "4x4x16"],
131
+ supports_accelerator_network_profile=False,
125
132
  docker_platform=DockerPlatform.AMD,
126
133
  sub_slicing_topologies=set(["4x4x8", "4x4x16"]),
127
134
  )
@@ -138,6 +145,7 @@ def test_get_tpu_system_characteristics_map_sets_super_slicing_support():
138
145
  gke_accelerator="test",
139
146
  machine_type="test",
140
147
  supported_topologies=["4x4x4", "4x4x8", "4x4x16"],
148
+ supports_accelerator_network_profile=False,
141
149
  docker_platform=DockerPlatform.AMD,
142
150
  super_slicing_topologies=set(["4x4x8", "4x4x16"]),
143
151
  )
@@ -154,6 +162,7 @@ def test_get_tpu_system_characteristics_map_prefers_default_topologies():
154
162
  gke_accelerator="test",
155
163
  machine_type="test",
156
164
  supported_topologies=["4x4x4", "4x4x32", "4x8x16", "8x8x8"],
165
+ supports_accelerator_network_profile=False,
157
166
  docker_platform=DockerPlatform.AMD,
158
167
  default_topologies=set(["4x8x16"]),
159
168
  )
@@ -206,6 +215,7 @@ def test_system_characteristics_post_init_sets_workload_policy_for_gpu():
206
215
  device_type="l4-1",
207
216
  supports_sub_slicing=False,
208
217
  supports_super_slicing=False,
218
+ supports_accelerator_network_profile=False,
209
219
  docker_platform=DockerPlatform.AMD,
210
220
  gpu_config=GpuConfig(requires_topology=False),
211
221
  )
@@ -225,5 +235,6 @@ def test_system_characteristics_post_init_throws_for_gpu_without_config():
225
235
  device_type="l4-1",
226
236
  supports_sub_slicing=False,
227
237
  supports_super_slicing=False,
238
+ supports_accelerator_network_profile=False,
228
239
  docker_platform=DockerPlatform.AMD,
229
240
  )
xpk/parser/common.py CHANGED
@@ -144,23 +144,6 @@ def add_cluster_arguments(
144
144
  )
145
145
 
146
146
 
147
- def add_kind_cluster_arguments(
148
- custom_parser_or_group: ParserOrArgumentGroup,
149
- ) -> None:
150
- """Add kind cluster arguments to the parser or argument group.
151
-
152
- Args:
153
- custom_parser_or_group: parser or argument group to add shared arguments to.
154
- """
155
- custom_parser_or_group.add_argument(
156
- '--kind-cluster',
157
- type=bool,
158
- action=argparse.BooleanOptionalAction,
159
- default=False,
160
- help='Apply command to a local test cluster.',
161
- )
162
-
163
-
164
147
  def add_global_arguments(custom_parser_or_group: ParserOrArgumentGroup):
165
148
  """Add global - no cloud dependent - arguments to the parser.
166
149
 
xpk/parser/core.py CHANGED
@@ -24,7 +24,6 @@ from .inspector import set_inspector_parser
24
24
  from .storage import set_storage_parser
25
25
  from .workload import set_workload_parsers
26
26
  from .info import set_info_parser
27
- from .kind import set_kind_parser
28
27
  from .version import set_version_parser
29
28
 
30
29
 
@@ -50,10 +49,6 @@ def set_parser(parser: argparse.ArgumentParser):
50
49
  "info",
51
50
  help="Commands around listing kueue clusterqueues and localqueues.",
52
51
  )
53
- kind_parser = xpk_subcommands.add_parser(
54
- "kind",
55
- help="commands around Kind cluster management",
56
- )
57
52
  version_parser = xpk_subcommands.add_parser(
58
53
  "version", help="Command to get xpk version"
59
54
  )
@@ -79,7 +74,6 @@ def set_parser(parser: argparse.ArgumentParser):
79
74
  workload_parser.print_help()
80
75
  info_parser.print_help()
81
76
  version_parser.print_help()
82
- kind_parser.print_help()
83
77
  config_parser.print_help()
84
78
 
85
79
  storage_parser.print_help()
@@ -89,7 +83,6 @@ def set_parser(parser: argparse.ArgumentParser):
89
83
  workload_parser.set_defaults(func=default_subcommand_function)
90
84
  cluster_parser.set_defaults(func=default_subcommand_function)
91
85
  info_parser.set_defaults(func=default_subcommand_function)
92
- kind_parser.set_defaults(func=default_subcommand_function)
93
86
  storage_parser.set_defaults(func=default_subcommand_function)
94
87
  version_parser.set_defaults(func=default_subcommand_function)
95
88
  config_parser.set_defaults(func=default_subcommand_function)
@@ -98,7 +91,6 @@ def set_parser(parser: argparse.ArgumentParser):
98
91
  set_cluster_parser(cluster_parser=cluster_parser)
99
92
  set_inspector_parser(inspector_parser=inspector_parser)
100
93
  set_info_parser(info_parser=info_parser)
101
- set_kind_parser(kind_parser=kind_parser)
102
94
  set_storage_parser(storage_parser=storage_parser)
103
95
  set_version_parser(version_parser=version_parser)
104
96
  set_config_parsers(config_parser=config_parser)
xpk/parser/storage.py CHANGED
@@ -25,7 +25,6 @@ from ..commands.storage import (
25
25
  )
26
26
  from .common import (
27
27
  add_cluster_arguments,
28
- add_kind_cluster_arguments,
29
28
  add_shared_arguments,
30
29
  )
31
30
  from typing import Protocol, Any
@@ -185,7 +184,6 @@ def add_storage_attach_parser(
185
184
  help='Comma-separated list of mountOptions for PersistentVolume',
186
185
  default='implicit-dirs',
187
186
  )
188
- add_kind_cluster_arguments(opt_args)
189
187
 
190
188
 
191
189
  def add_storage_create_parser(storage_subcommands_parser: Subcommands) -> None:
@@ -284,8 +282,6 @@ def add_storage_create_parser(storage_subcommands_parser: Subcommands) -> None:
284
282
  default='',
285
283
  )
286
284
 
287
- add_kind_cluster_arguments(opt_args)
288
-
289
285
 
290
286
  def add_storage_list_parser(storage_subcommands_parser: Subcommands) -> None:
291
287
  storage_list_parser: argparse.ArgumentParser = (
@@ -319,12 +315,6 @@ def add_storage_detach_parser(storage_subcommands_parser: Subcommands) -> None:
319
315
  req_args.add_argument('name', type=str)
320
316
  add_cluster_arguments(req_args, required=True)
321
317
 
322
- opt_args = storage_detach_parser.add_argument_group(
323
- 'Optional Arguments',
324
- 'Optional arguments for storage delete.',
325
- )
326
- add_kind_cluster_arguments(opt_args)
327
-
328
318
 
329
319
  def add_storage_delete_parser(storage_subcommands_parser: Subcommands) -> None:
330
320
  storage_delete_parser: argparse.ArgumentParser = (
@@ -352,4 +342,3 @@ def add_storage_delete_parser(storage_subcommands_parser: Subcommands) -> None:
352
342
  action='store_true',
353
343
  help='Force filestore instance deletion even if it has attached storages',
354
344
  )
355
- add_kind_cluster_arguments(opt_args)
@@ -28,7 +28,7 @@ def _get_boolean_flag(flag: str, default: bool) -> bool:
28
28
 
29
29
  class _FeatureFlags:
30
30
  SUB_SLICING_ENABLED = _get_boolean_flag("SUB_SLICING_ENABLED", default=False)
31
- TELEMETRY_ENABLED = _get_boolean_flag("TELEMETRY_ENABLED", default=False)
31
+ TELEMETRY_ENABLED = _get_boolean_flag("TELEMETRY_ENABLED", default=True)
32
32
  SUPER_SLICING_ENABLED = _get_boolean_flag(
33
33
  "SUPER_SLICING_ENABLED", default=False
34
34
  )
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: xpk
3
- Version: 0.17.3
3
+ Version: 1.0.0
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
@@ -114,10 +114,21 @@ XPK also supports the following [Google Cloud Storage solutions](./docs/usage/st
114
114
  * [Storage](./docs/usage/storage.md)
115
115
  * [Advanced](./docs/usage/advanced.md)
116
116
  * [Inspector](./docs/usage/inspector.md)
117
- * [Run](./docs/usage/run.md)
118
- * [Job](./docs/usage/job.md)
119
117
  * [Troubleshooting](./docs/troubleshooting.md)
120
- * [Local Testing](./docs/local_testing.md)
118
+
119
+ # Privacy notice
120
+
121
+ To help improve XPK, feature usage statistics are collected and sent to Google. You can opt-out at any time by executing
122
+ the following shell command:
123
+
124
+ ```shell
125
+ xpk config set send-telemetry <true/false>
126
+ ```
127
+
128
+ XPK telemetry overall is handled in accordance with the [Google Privacy Policy](https://policies.google.com/privacy). When
129
+ you use XPK to interact with or utilize GCP Services, your information is handled in accordance with the
130
+ [Google Cloud Privacy Notice](https://cloud.google.com/terms/cloud-privacy-notice).
131
+
121
132
 
122
133
  # Contributing
123
134
 
@@ -20,29 +20,28 @@ xpk/blueprints/a4/config-map.yaml.tftpl,sha256=o6LeGIYUfFGyj3vj-8ztV5ildQ46QZVl7
20
20
  xpk/blueprints/a4/nccl-rdma-installer-a4.yaml,sha256=if3WOmNLVGTJIJHU76EWC1FyiIXDTRIXcwo4OsBxarQ,2113
21
21
  xpk/blueprints/a4/storage_crd.yaml,sha256=r4WFXnSJJ25EUF-t4Ljfbl-cJoSaiFiZkP8451eTub4,1260
22
22
  xpk/commands/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
23
- xpk/commands/cluster.py,sha256=5ebvHXe8Bi4haMd1RokJbjP0LemqXxC1EseqOLWIkGw,45659
23
+ xpk/commands/cluster.py,sha256=d5L8Kqfk93SZCxbPCZ5oePkqruWYZKi49rJ6yGgZSMg,45479
24
24
  xpk/commands/cluster_gcluster.py,sha256=Ig8jLjsiyFgw9U4BBEzDK2diA9m0STKQgz-uUTG_vYE,13731
25
- xpk/commands/cluster_gcluster_test.py,sha256=s1wwkcdY4LTxmk_Tx5PKdh9pZmEoo1n8XlzybWalc0M,6165
25
+ xpk/commands/cluster_gcluster_test.py,sha256=zdxz5gAMu3HRVNsj7F-VYRf4TYSPMjuOG7DolQN2Pb4,6263
26
26
  xpk/commands/cluster_test.py,sha256=aMkwKrhoEuqElME16ztx5lwv4zT0z_xV0L3in1RaW6M,24017
27
27
  xpk/commands/common.py,sha256=p43sspD5RfYRj3Se_b-X0s0dbBs1PMI1qtySg6zZKKg,2706
28
28
  xpk/commands/config.py,sha256=L_zRpQTxMcSh6rxOT8gG263V6YGqzVoz4UxdWywTFdA,850
29
29
  xpk/commands/info.py,sha256=uhv5mPfgg9N-5JhQw4dT2jujL9ZC5kzGA18h9NFfm5A,7429
30
30
  xpk/commands/inspector.py,sha256=FPasKtGuEZKNXIQin4AG49clfD4b53NxXpWqBPZIIoE,12955
31
- xpk/commands/kind.py,sha256=ck4zaJh9kaNluEdBew9OVj4cZXjUYk_ElycqGLo6f7g,7355
32
31
  xpk/commands/managed_ml_diagnostics.py,sha256=87wmFbnYQY-kEpJfPo1Up53xM5P_P5wOlXczxHzxJjQ,6984
33
32
  xpk/commands/managed_ml_diagnostics_test.py,sha256=pQ1YUGMGRQFJYTS_1o9YyGUzYdLaBdA84LjbnncaeEo,3828
34
33
  xpk/commands/storage.py,sha256=cSTJN9Mjvdsvk_Nk43kVdQFhp89nxWbanDsTOGZCkpQ,10708
35
34
  xpk/commands/version.py,sha256=k30rdLP9clUM8eeSwRFhpfzSb1qwcQImTfuC59Ed6CA,771
36
- xpk/commands/workload.py,sha256=l99NRFLs7pXuaLdn5d-Pid-cZulKpB3FNus-HdNDtZw,31513
37
- xpk/commands/workload_test.py,sha256=iXTY7VR1KrlPZZyh1Zm0N946kIP1iV2Fnqx1NtOYDJU,7274
35
+ xpk/commands/workload.py,sha256=HV54qdjxR4fceQjae9Qpgdk9BY2C5Wh8a9aOS_kFY4E,31725
36
+ xpk/commands/workload_test.py,sha256=KLSI4L01fWmPm7xNPXEpjABU5FC9P3CXVto1ifSiqKY,7322
38
37
  xpk/core/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
39
38
  xpk/core/capacity.py,sha256=_TyWayBkNU8fBpz1LTbCddEFZiZW5Qz-xmJnQMsXh0c,10534
40
39
  xpk/core/capacity_test.py,sha256=jZjMHTYlFLdAmBN1t9k29iABCSE5hlW0--q7QLDQpfQ,4330
41
- xpk/core/cluster.py,sha256=3nl77I_MgQpBZsZSzsiQ_7IyFRzfLrYNRUL1gsSNhKU,24036
40
+ xpk/core/cluster.py,sha256=_h82ZmYufvBOezEUvy-DWOJv9h3SQsy5isL9qbYdq44,24001
42
41
  xpk/core/cluster_private.py,sha256=RLi0C7bV0NEUXl6QKQzvUT0weN9EdqPvjuuOQsNO0DY,6868
43
42
  xpk/core/cluster_test.py,sha256=J4Wk7E--ik_IsWWzL_iWGWbx99Ih03m-0bs-uU7gGDg,5853
44
43
  xpk/core/commands.py,sha256=at73VJHdZ4rVA8uvW997tNrvnCjP9v6zaw96bU0kd74,10841
45
- xpk/core/config.py,sha256=7U8jI5oZcgV_UnOHSS3huUIlDmPNREM-ml0N1Y9IvGM,4612
44
+ xpk/core/config.py,sha256=U2JDXx-XBuqQpZJf2iUDoww5--E8ejZfgmIxKeGu-gU,4668
46
45
  xpk/core/config_test.py,sha256=POSuofK0LFbNNygDAo2fjtKY4NMrRjUFeGcpBh9JOS4,3569
47
46
  xpk/core/docker_container.py,sha256=8hqWWNKtjf6dqCFRpfndTMGvN_NS6zhfBr7YuKfh7qo,7626
48
47
  xpk/core/docker_image.py,sha256=9vwqbb6Mc3C5ZEOph03WS-EWI5hxMYGGigqzIMkDTjE,6909
@@ -55,22 +54,22 @@ xpk/core/gcluster_manager.py,sha256=lyv_MvdnkByy9_PEBj_ugAEBwnCbFNiWTSrEFjrMlPc,
55
54
  xpk/core/gcsfuse.py,sha256=kg5pgxdTjgiqquuGjev9fXzJPb8oiWPTK6wzCddzheQ,2125
56
55
  xpk/core/jobset.py,sha256=PJ4Fd8TNNLuYKNOMehoMYRIUEXyc5jsbHctJGqfW_8Y,4037
57
56
  xpk/core/kueue_manager.py,sha256=JB8DcD-RFvBdC9Mk_DDCAkI2Km8W5-KMTRMVec06LlM,20010
58
- xpk/core/kueue_manager_test.py,sha256=FfBd1vninU_fcJ9wZev45-vpEsH12a9-XKysk_h4auo,22008
57
+ xpk/core/kueue_manager_test.py,sha256=ZYnIOFN2ZgnrxBhoBkh9JUXP5YbFsoFI11iQnHspafI,22109
59
58
  xpk/core/monitoring.py,sha256=__bzTq_DIDAK8yIaN4F3MJh-yjYw5X1OlxmRgYOpf1g,4332
60
59
  xpk/core/mtc.py,sha256=pO7p3l-EzLFdTE8MdwWV8i0Zu-7epGql_kPoksVofIU,6259
61
60
  xpk/core/nap.py,sha256=7haJtWVfe9csfK-LmmIcDnmXCPIJFnoSviHaZ4y4i6s,12556
62
61
  xpk/core/network.py,sha256=Oulb7U69lWkpOKxOC1C7ekJDpC51TLwd7XdZA3NQ7E0,10505
63
- xpk/core/nodepool.py,sha256=X8sANyzpDoYhV5uWfEYV0RnHfvj_o1G84hArqWaNDu8,24301
64
- xpk/core/nodepool_test.py,sha256=k29tg-lDNEZP6KUNJQE0sHIKFLD6LJ1PSLVtWHzqh5E,13161
62
+ xpk/core/nodepool.py,sha256=ehncCzvtDZ5lhyejRkYQCBjlcKHwKh3PXuGSLY1eWz8,24504
63
+ xpk/core/nodepool_test.py,sha256=qx0VQ_YpgjB8Sf1KJvQCKQuCd3NmWvjBQuKdnT8NWKU,13358
65
64
  xpk/core/pathways.py,sha256=32GxCIPiEBqSpK6g2gMmB7Nxj_HlG4I30u1C9UyWl1A,11594
66
65
  xpk/core/pathways_test.py,sha256=UeuSo_g9BNI27to-wflQwc6dJFVSA5-kOK_cjmY5qgU,1809
67
66
  xpk/core/ray.py,sha256=JWhc_ToRHpF4_URGnuE_47FMgamaRsA4KVUMpqThWzw,6145
68
67
  xpk/core/resources.py,sha256=dDsG_LOtcU17p1UKgOYyjdPxbMfqcb7pJ4SjfLDA6Os,9389
69
- xpk/core/scheduling.py,sha256=RMoei_HUs03rfrEC-HYk7ONzg9BRKwr59-KljCR2TMo,11560
70
- xpk/core/scheduling_test.py,sha256=iYnzXv_MjN743pa4zYAgRqb-6dB9nVPpLI7JP5S8M2I,14463
68
+ xpk/core/scheduling.py,sha256=ucNOidEO_QQekIL44zG6Yhpr8gErPjJa6B2JJHb7diY,12404
69
+ xpk/core/scheduling_test.py,sha256=dzdrUCc61Twu7WC86t1XNDg-7bWLYYz4hqBDWIVHKL4,15883
71
70
  xpk/core/storage.py,sha256=NILvVAcLNMLmp4wKx_TEKbMMF5X1oL-FrQV46PT0_ds,16902
72
- xpk/core/system_characteristics.py,sha256=ZQbTbjaeT3Q12kmobz14U878w3FWnXDCetiLZQlVAdY,32127
73
- xpk/core/system_characteristics_test.py,sha256=sREN8u8bC0ze_q9hY3v-ZxC7so-_Ox1mt_DkIbUgHJ4,7477
71
+ xpk/core/system_characteristics.py,sha256=y9v4WRN-u9yvo990MKbWTUOWlzn-f6vfaMNeikQlhHY,34115
72
+ xpk/core/system_characteristics_test.py,sha256=XVaKJ5wYdNwwwUKBnuK3zd1u-Qj3VnJR7MHlOeCa-K0,8029
74
73
  xpk/core/telemetry.py,sha256=R7IONNl5heMoNcOurfT3I34XJrBEODKVY88ONiDGuqE,7512
75
74
  xpk/core/telemetry_test.py,sha256=ll-B1ut9X-por17fpQnNb6hKrfyoZanMWRPbvqWrXss,8261
76
75
  xpk/core/updates.py,sha256=FxLDgEL2O-qnslhT9U60NG5gzXmSv8Fn2wPUf3YZLM8,1734
@@ -102,14 +101,13 @@ xpk/core/workload_decorators/tcpxo_decorator.py,sha256=5SgL-7aTHclN7rvCGvEOjZoUi
102
101
  xpk/parser/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
103
102
  xpk/parser/cluster.py,sha256=U2T-Q4yS86PWeFLNfknYWDDzZfubCKqIhqasxKLmErI,31342
104
103
  xpk/parser/cluster_test.py,sha256=xzQEC3IeAMpwsbNbHLuaNKxR3iaZcm3z4m3i61G62d4,6581
105
- xpk/parser/common.py,sha256=sJYGjrn2YgFxelDCYB18s1R8Md8GpDcMQNoAezxDDIs,7257
104
+ xpk/parser/common.py,sha256=DBj0MQHbxcquPWJ3WcwdiaKhGJZgjdppNJrb9iUFQsE,6797
106
105
  xpk/parser/common_test.py,sha256=_6Fm2pUF7h4K0G5qxGabXSYr4ng9ihOzlViE6oLQwQs,1557
107
106
  xpk/parser/config.py,sha256=-XnWx9aFsBW4Uzo_hpOMD2ZQ0bdZLvq1ksv83_5jqSM,1633
108
- xpk/parser/core.py,sha256=P2Dx3AbTlDoWnCCrMhVdr3Fs5FEzYFlmSiugkun1GL0,3623
107
+ xpk/parser/core.py,sha256=rUgPYrqBgqoeuVVwcu3qMABs5KZ3jZI-lfkKywwpGYo,3340
109
108
  xpk/parser/info.py,sha256=UJohxVVWdt9IgUXoPsrVae2DN1BjAVGWrSN2ajrB8RQ,1860
110
109
  xpk/parser/inspector.py,sha256=hAPAZ2k9iSJgC1mjnz3rMleInsAQ8PmkyyUKFyBmsgY,1997
111
- xpk/parser/kind.py,sha256=sgPCqNVrgmFLcOBEbhlaphwVXxMh_opP9ntCq4KPePE,2682
112
- xpk/parser/storage.py,sha256=0V1d1htsjoa-SuxOX_vNxz2Lg4Nue9CBe_H0bNS2Hv0,10270
110
+ xpk/parser/storage.py,sha256=VnMWSGW1xNIMz_cU-dk1CtpGSZEtZ9ecz1KI3bpQPqw,9945
113
111
  xpk/parser/storage_test.py,sha256=i_F9cuQXHRvUy4RJwbfuuI8ZVpTpkkY96sZ1GZ4dLPw,1494
114
112
  xpk/parser/validators.py,sha256=-NBZelvfwZRzjz-YUCreD8EzMLHll8PZM-d-MVm2PG4,1192
115
113
  xpk/parser/version.py,sha256=eJo4PAbbmRQZulgKBs_ytbVgV9zAaaXeNzMMxmgFMVY,769
@@ -133,7 +131,7 @@ xpk/utils/__init__.py,sha256=YPwWBbgLAu7L-YlTVGB2r8ZV4TzypURMRBcehSHHlLY,561
133
131
  xpk/utils/console.py,sha256=AJWSyjuWyLjb7SYt8kPb0gw9N84EN9LbLxYCXjC-6Ds,2464
134
132
  xpk/utils/console_test.py,sha256=x1v7v9VrIZwAKH-eOzj1lAY4EsHxJ6ruhfEOzpssO6o,2944
135
133
  xpk/utils/execution_context.py,sha256=hONGz1hQSKE-puah2rE_uN9YUeEC4oW82VOryw5_Vgo,1181
136
- xpk/utils/feature_flags.py,sha256=lRNjJIKyMUvUkF35MjG0iQYuoxSZarf-W98gWOW8r6M,1161
134
+ xpk/utils/feature_flags.py,sha256=9QSMpdxcGR84unhOGJyCteT9R92-h5K6tOcjn5YHgDw,1160
137
135
  xpk/utils/file.py,sha256=yB1-k3FahoxkBpojB59vQNeZYOXB3pmktnjU4Ceah7M,2605
138
136
  xpk/utils/gcs_utils.py,sha256=zg-XSTv4G4TFjeT2bNBm2WLdDXPrOZi0rNv_JdppNg4,4113
139
137
  xpk/utils/kubectl.py,sha256=WKB9UhpouPN9G4n2ejRi_PgsYLI0R01gzkS1WGU6mJA,1828
@@ -151,9 +149,9 @@ xpk/utils/validation.py,sha256=rE9LTkXJT7jIesodFb9pONL7ixhLqiQleyoaz7N39Dw,2765
151
149
  xpk/utils/validation_test.py,sha256=PEDSMUqZdt_Lx1FSR-LOTXKKtsJ47JH1fxugM0Gfz6Y,1168
152
150
  xpk/utils/versions.py,sha256=_Ep68W70a9605XjiaOOpBa9Is9jXlsoOiwL8v5Xt-WA,897
153
151
  xpk/utils/yaml.py,sha256=j8xuAJ9yAAwnQi6ozwZ-nMnDyDnc3xWkeBZMtSuP4RU,844
154
- xpk-0.17.3.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
155
- xpk-0.17.3.dist-info/METADATA,sha256=ONK-6JpzJboT0wF60svxA4SKJTSqeQ4KNfgSvZ_kkDY,7930
156
- xpk-0.17.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
157
- xpk-0.17.3.dist-info/entry_points.txt,sha256=mzEtiIesFkT1kmcTUVDA1o3uOhiniX6tIz2wmOlMu1M,38
158
- xpk-0.17.3.dist-info/top_level.txt,sha256=TQKZWgV7LSElvmunYT9V_627qOMoxq3qYzWAFzKudB8,16
159
- xpk-0.17.3.dist-info/RECORD,,
152
+ xpk-1.0.0.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
153
+ xpk-1.0.0.dist-info/METADATA,sha256=9woUe1dyR3fKZPXpoJeLwWTSNKCPZO--_LWee7UPRNc,8364
154
+ xpk-1.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
155
+ xpk-1.0.0.dist-info/entry_points.txt,sha256=mzEtiIesFkT1kmcTUVDA1o3uOhiniX6tIz2wmOlMu1M,38
156
+ xpk-1.0.0.dist-info/top_level.txt,sha256=TQKZWgV7LSElvmunYT9V_627qOMoxq3qYzWAFzKudB8,16
157
+ xpk-1.0.0.dist-info/RECORD,,