xpk 0.17.1__py3-none-any.whl → 0.17.3__py3-none-any.whl

xpk/commands/cluster.py

@@ -49,7 +49,6 @@ from ..core.gcloud_context import (
     zone_to_region,
 )
 from ..core.jobset import update_jobset_resources_if_necessary
-from ..core.kjob import apply_kjob_crds, prepare_kjob, verify_kjob_installed
 from ..core.kueue_manager import (KueueConfig, KueueManager)
 from ..core.nap import enable_autoprovisioning_on_cluster
 from ..core.network import (
@@ -98,7 +97,6 @@ def cluster_adapt(args) -> None:
   if should_validate_dependencies(args):
     validate_dependencies_list([
         SystemDependency.KUBECTL,
-        SystemDependency.KJOB,
         SystemDependency.GCLOUD,
     ])
   args.enable_pathways = False
@@ -188,7 +186,6 @@ def cluster_adapt(args) -> None:
   if install_kueue_code != 0:
     xpk_exit(install_kueue_code)
 
-  install_kjob(args)
   if system.accelerator_type == AcceleratorType.GPU:
     prepare_gpus(system)
 
@@ -308,7 +305,6 @@ def cluster_create(args) -> None:
   if should_validate_dependencies(args):
     validate_dependencies_list([
         SystemDependency.KUBECTL,
-        SystemDependency.KJOB,
         SystemDependency.GCLOUD,
     ])
 
@@ -455,8 +451,6 @@ def cluster_create(args) -> None:
   if install_kueue_code != 0:
     xpk_exit(install_kueue_code)
 
-  install_kjob(args)
-
   if system.accelerator_type == AcceleratorType.GPU:
     prepare_gpus(system)
 
@@ -1343,22 +1337,6 @@ def install_storage_csis(args):
       xpk_exit(update_cluster_command_code)
 
 
-def install_kjob(args):
-  xpk_print('Verifying kjob installation')
-  err_code = verify_kjob_installed()
-  if err_code > 0:
-    xpk_exit(err_code)
-
-  xpk_print('Applying kjob CDRs')
-  err_code = apply_kjob_crds()
-  if err_code > 0:
-    xpk_exit(err_code)
-
-  err_code = prepare_kjob(args)
-  if err_code > 0:
-    xpk_exit(err_code)
-
-
 def _install_kueue(
     args,
     system: SystemCharacteristics,

xpk/commands/cluster_gcluster.py

@@ -38,7 +38,6 @@ from ..core.commands import run_command_for_value
 from ..core.docker_manager import DockerManager
 from ..core.gcloud_context import zone_to_region
 from ..core.gcluster_manager import GclusterManager
-from ..core.kjob import apply_kjob_crds, prepare_kjob
 from ..core.remote_state.fuse_remote_state import FuseStateClient
 from ..core.remote_state.remote_state_client import RemoteStateClient
 from ..utils.console import xpk_exit, xpk_print
@@ -112,18 +111,7 @@ def cluster_create(
   get_cluster_credentials(args)
 
   err_code = __install_kueue(args)
-  if err_code > 0:
-    xpk_exit(err_code)
-
-  err_code = apply_kjob_crds()
-  if err_code > 0:
-    xpk_exit(err_code)
-
-  err_code = prepare_kjob(args)
-  if err_code > 0:
-    xpk_exit(err_code)
-
-  xpk_exit(0)
+  xpk_exit(err_code)
 
 
 def __install_kueue(args) -> int:

xpk/commands/cluster_gcluster_test.py

@@ -46,8 +46,6 @@ def mock_cluster_create_deps(request):
   """Mocks dependencies for cluster_create."""
   with (
       patch("xpk.commands.cluster_gcluster.xpk_exit") as mock_exit,
-      patch("xpk.commands.cluster_gcluster.prepare_kjob") as mock_prep_kjob,
-      patch("xpk.commands.cluster_gcluster.apply_kjob_crds") as mock_apply_kjob,
       patch(
           "xpk.commands.cluster_gcluster.get_cluster_credentials"
       ) as mock_get_creds,
@@ -68,8 +66,6 @@ def mock_cluster_create_deps(request):
   ):
     yield {
         "xpk_exit": mock_exit,
-        "prepare_kjob": mock_prep_kjob,
-        "apply_kjob_crds": mock_apply_kjob,
         "get_cluster_credentials": mock_get_creds,
         "generate_blueprint": mock_gen_bp,
         "prepare_gcluster_manager": mock_prep_gcm,
@@ -85,9 +81,6 @@ def test_install_kueue_standard(
     mock_get_total_chips, mock_args, mock_cluster_create_deps
 ):
   """Tests __install_kueue for a standard installation."""
-  mock_cluster_create_deps["prepare_kjob"].return_value = 0
-  mock_cluster_create_deps["apply_kjob_crds"].return_value = 0
-
   mock_system = SystemCharacteristics(
       topology="N/A",
       vms_per_slice=1,
@@ -138,9 +131,6 @@ def test_install_kueue_with_autoprovisioning(
     mock_enable_autoprovisioning, mock_args, mock_cluster_create_deps
 ):
   """Tests __install_kueue with autoprovisioning enabled."""
-  mock_cluster_create_deps["prepare_kjob"].return_value = 0
-  mock_cluster_create_deps["apply_kjob_crds"].return_value = 0
-
   mock_args.enable_autoprovisioning = True
   mock_system = SystemCharacteristics(
       topology="N/A",

xpk/commands/cluster_test.py

@@ -56,7 +56,6 @@ class _ClusterCreateMocks:
   create_cluster_configmaps: MagicMock
   set_jobset_on_cluster: MagicMock
   get_cluster_location: MagicMock
-  install_kjob: MagicMock
   xpk_exit: MagicMock
   update_jobset_resources_if_necessary: MagicMock
   _install_kueue: MagicMock
@@ -204,9 +203,6 @@ def cluster_create_mocks(mocker) -> _ClusterCreateMocks:
           'xpk.commands.cluster.get_cluster_location',
           return_value='us-central1',
       ),
-      install_kjob=mocker.patch(
-          'xpk.commands.cluster.install_kjob', return_value=0
-      ),
       xpk_exit=mocker.patch('xpk.commands.cluster.xpk_exit'),
       update_jobset_resources_if_necessary=mocker.patch(
           'xpk.commands.cluster.update_jobset_resources_if_necessary',

xpk/commands/kind.py

@@ -20,11 +20,6 @@ from ..core.commands import (
     run_command_with_updates,
 )
 from ..core.cluster import set_jobset_on_cluster, setup_k8s_env
-from ..core.kjob import (
-    verify_kjob_installed,
-    prepare_kjob,
-    apply_kjob_crds,
-)
 from ..core.scheduling import get_total_chips_requested_from_args
 from ..core.storage import install_storage_crd
 from ..core.system_characteristics import (
@@ -48,7 +43,6 @@ def cluster_create(args) -> None:
   if should_validate_dependencies(args):
     validate_dependencies_list([
         SystemDependency.KUBECTL,
-        SystemDependency.KJOB,
         SystemDependency.GCLOUD,
     ])
   xpk_print(f'Starting cluster create for cluster {args.cluster}:', flush=True)
@@ -69,21 +63,6 @@ def cluster_create(args) -> None:
   if set_jobset_on_cluster_code != 0:
     xpk_exit(set_jobset_on_cluster_code)
 
-  xpk_print('Verifying kjob installation')
-  err_code = verify_kjob_installed()
-  if err_code > 0:
-    xpk_exit(err_code)
-
-  xpk_print('Applying kjob CDRs')
-  err_code = apply_kjob_crds()
-  if err_code > 0:
-    xpk_exit(err_code)
-
-  args.kind_cluster = True
-  err_code = prepare_kjob(args)
-  if err_code > 0:
-    xpk_exit(err_code)
-
   k8s_client = setup_k8s_env(args)
   install_storage_crd(k8s_client)
 

xpk/commands/storage.py

@@ -23,7 +23,6 @@ from kubernetes.client.rest import ApiException
 
 from ..core import gcsfuse
 from ..core.cluster import (
-    DEFAULT_NAMESPACE,
     add_zone_and_project,
     get_cluster_network,
     setup_k8s_env,
@@ -35,12 +34,6 @@ from ..core.cluster import (
     update_cluster_with_workload_identity_if_necessary,
 )
 from ..core.filestore import FilestoreClient, get_storage_class_name
-from ..core.kjob import (
-    KJOB_API_GROUP_NAME,
-    KJOB_API_GROUP_VERSION,
-    KJOB_API_VOLUME_BUNDLE_PLURAL,
-    create_volume_bundle_instance,
-)
 from ..core.storage import (
     GCP_FILESTORE_TYPE,
     GCS_FUSE_TYPE,
@@ -98,9 +91,6 @@ def storage_create(args: Namespace) -> None:
 
     k8s_api_client = setup_k8s_env(args)
     create_storage_crds(k8s_api_client, args, manifest)
-    create_volume_bundle_instance(
-        k8s_api_client, args.name, manifest, args.readonly, args.mount_point
-    )
     # Not required for Filestore. Will be uncommented when adding GCSFuse create
     # return_code = update_cluster_with_workload_identity_if_necessary(args)
     # if return_code > 0:
@@ -214,9 +204,6 @@ def storage_attach(args: Namespace) -> None:
 
   k8s_api_client = setup_k8s_env(args)
   create_storage_crds(k8s_api_client, args, manifest)
-  create_volume_bundle_instance(
-      k8s_api_client, args.name, manifest, args.readonly, args.mount_point
-  )
 
   enable_csi_drivers_if_necessary(args)
 
@@ -332,18 +319,6 @@ def delete_storage_resources(k8s_api_client: ApiClient, storage: Storage):
         "Storage Class",
     )
 
-  delete_resource(
-      lambda name: api_instance.delete_namespaced_custom_object(
-          namespace=DEFAULT_NAMESPACE,
-          name=name,
-          group=KJOB_API_GROUP_NAME,
-          version=KJOB_API_GROUP_VERSION,
-          plural=KJOB_API_VOLUME_BUNDLE_PLURAL,
-      ),
-      storage.name,
-      "VolumeBundle",
-  )
-
   delete_resource(
       lambda name: api_instance.delete_cluster_custom_object(
           name=name,

xpk/core/cluster.py

@@ -717,10 +717,8 @@ def get_cluster_credentials(args) -> int:
       location=location,
       dns_endpoint=True,
   )
-  if return_code != 0:
-    return return_code
 
-  if not _are_credentials_valid():
+  if return_code != 0 or not _are_credentials_valid():
     xpk_print('Detected error. Retrying without --dns-endpoint flag...')
     return_code = _get_credentials(
         project=args.project,

xpk/core/config.py

@@ -53,14 +53,6 @@ PROJECT_KEY = 'project-id'
 CLIENT_ID_KEY = 'client-id'
 SEND_TELEMETRY_KEY = 'send-telemetry'
 ZONE_KEY = 'zone'
-KJOB_BATCH_IMAGE = 'batch-image'
-KJOB_BATCH_WORKING_DIRECTORY = 'batch-working-directory'
-KJOB_SHELL_IMAGE = 'shell-image'
-KJOB_SHELL_INTERACTIVE_COMMAND = 'shell-interactive-command'
-KJOB_SHELL_WORKING_DIRECTORY = 'shell-working-directory'
-CONFIGS_KEY = 'configs'
-GKE_ENDPOINT_KEY = 'gke-endpoint'
-DEPENDENCIES_KEY = 'deps-verified-version'
 
 DEFAULT_KEYS = [
     CFG_BUCKET_KEY,
@@ -69,13 +61,6 @@ DEFAULT_KEYS = [
     CLIENT_ID_KEY,
     SEND_TELEMETRY_KEY,
     ZONE_KEY,
-    GKE_ENDPOINT_KEY,
-    DEPENDENCIES_KEY,
-    KJOB_BATCH_IMAGE,
-    KJOB_BATCH_WORKING_DIRECTORY,
-    KJOB_SHELL_IMAGE,
-    KJOB_SHELL_INTERACTIVE_COMMAND,
-    KJOB_SHELL_WORKING_DIRECTORY,
 ]
 VERTEX_TENSORBOARD_FEATURE_FLAG = XPK_CURRENT_VERSION >= '0.4.0'
 

xpk/core/system_characteristics.py

@@ -80,15 +80,6 @@ class GpuConfig:
 
   requires_topology: bool
   gpu_direct_name: Literal['fastrak', 'rdma', 'tcpx', 'tcpxo'] = 'fastrak'
-  kjob_decorator_fn: Optional[Callable[[dict], dict]] = None
-  """A function to decorate the kjob template for GPU-specific configurations.
-
-  Args:
-    job_manifest (dict): The kjob manifest as a dictionary.
-
-  Returns:
-    dict: The modified kjob manifest as a dictionary.
-  """
   nccl_installer: Optional[str] = None
   jobset_decorator_fn: Optional[Callable[[str, list[str]], str]] = None
   """A function to decorate the jobset for GPU-specific configurations.
@@ -106,7 +97,7 @@ class GpuConfig:
     parts = []
     for f in dataclasses.fields(self):
       value = getattr(self, f.name)
-      if f.name in ('kjob_decorator_fn', 'jobset_decorator_fn') and value:
+      if f.name in ('jobset_decorator_fn') and value:
         parts.append(f'{f.name}=<function {value.__name__}>')
       else:
         parts.append(f'{f.name}={repr(value)}')
@@ -420,7 +411,6 @@ UserFacingNameToSystemCharacteristics = {
         gpu_config=GpuConfig(
             requires_topology=True,
             nccl_installer=INSTALLER_NCCL_RDMA_A4X,
-            kjob_decorator_fn=rdma_decorator.decorate_kjob_template,
             jobset_decorator_fn=rdma_decorator.decorate_jobset,
             gpu_direct_name='rdma',
         ),
@@ -439,7 +429,6 @@ UserFacingNameToSystemCharacteristics = {
         gpu_config=GpuConfig(
             requires_topology=True,
             nccl_installer=INSTALLER_NCCL_RDMA_A4X,
-            kjob_decorator_fn=rdma_decorator.decorate_kjob_template,
             jobset_decorator_fn=rdma_decorator.decorate_jobset,
             gpu_direct_name='rdma',
         ),
@@ -458,7 +447,6 @@ UserFacingNameToSystemCharacteristics = {
         gpu_config=GpuConfig(
             requires_topology=True,
             nccl_installer=INSTALLER_NCCL_RDMA,
-            kjob_decorator_fn=rdma_decorator.decorate_kjob_template,
             jobset_decorator_fn=rdma_decorator.decorate_jobset,
             gpu_direct_name='rdma',
         ),
@@ -477,7 +465,6 @@ UserFacingNameToSystemCharacteristics = {
         gpu_config=GpuConfig(
             requires_topology=True,
             nccl_installer=INSTALLER_NCCL_RDMA,
-            kjob_decorator_fn=rdma_decorator.decorate_kjob_template,
             jobset_decorator_fn=rdma_decorator.decorate_jobset,
             gpu_direct_name='rdma',
         ),
@@ -497,7 +484,6 @@ UserFacingNameToSystemCharacteristics = {
         gpu_config=GpuConfig(
             requires_topology=True,
             nccl_installer=INSTALLER_NCCL_TCPX,
-            kjob_decorator_fn=tcpx_decorator.decorate_kjob_template,
             jobset_decorator_fn=tcpx_decorator.decorate_jobset,
             gpu_direct_name='tcpx',
         ),
@@ -517,7 +503,6 @@ UserFacingNameToSystemCharacteristics = {
         gpu_config=GpuConfig(
             requires_topology=True,
             nccl_installer=INSTALLER_NCCL_TCPXO,
-            kjob_decorator_fn=tcpxo_decorator.decorate_kjob_template,
             jobset_decorator_fn=tcpxo_decorator.decorate_jobset,
             gpu_direct_name='tcpxo',
         ),

xpk/core/workload_decorators/rdma_decorator.py

@@ -18,21 +18,6 @@ import yaml
 from ...utils.yaml import literal_string
 
 
-def decorate_kjob_template(job_manifest: dict) -> dict:
-  spec = (
-      job_manifest.setdefault('spec', {})
-      .setdefault('template', {})
-      .setdefault('spec', {})
-  )
-  spec.setdefault('tolerations', [])
-  spec.setdefault('volumes', [])
-
-  add_volumes(job_manifest)
-  add_tolerations(job_manifest)
-  update_gpu_containers(job_manifest)
-  return job_manifest
-
-
 def decorate_jobset(jobset_manifest_str: str, sub_networks: list[str]) -> str:
   """
   Decorates a JobSet manifest with the necessary components for rdma-daemon.

xpk/core/workload_decorators/tcpx_decorator.py

@@ -22,14 +22,6 @@ from ...utils.yaml import literal_string
 tcpx = 'v2.0.11'
 
 
-def decorate_kjob_template(job_manifest: dict) -> dict:
-  add_volumes(job_manifest)
-  add_tolerations(job_manifest)
-  add_tcpx_daemon_container(job_manifest)
-  update_gpu_containers(job_manifest)
-  return job_manifest
-
-
 def decorate_job(job_manifest: dict) -> dict:
   add_annotations(job_manifest)
   add_volumes(job_manifest)

xpk/core/workload_decorators/tcpx_decorator_test.py

@@ -47,24 +47,6 @@ spec:
                 image: my-sidecar-image
 """
 
-# Minimal kjob template for testing
-BASE_KJOB_TEMPLATE = {
-    "spec": {
-        "template": {
-            "spec": {
-                "containers": [
-                    {
-                        "name": "main-gpu-container",
-                        "image": "my-gpu-image",
-                        "resources": {"limits": {"nvidia.com/gpu": 8}},
-                    },
-                    {"name": "sidecar-container", "image": "my-sidecar-image"},
-                ]
-            }
-        }
-    }
-}
-
 # Minimal job manifest for testing
 BASE_JOB_MANIFEST = {
     "spec": {
@@ -205,63 +187,3 @@ def test_decorate_job():
   assert "devices.gke.io/container.tcpx-daemon" in annotations
   assert "networking.gke.io/default-interface" in annotations
   assert "networking.gke.io/interfaces" in annotations
-
-
-def test_decorate_kjob_template():
-  """Tests decorate_kjob_template."""
-  kjob_template = copy.deepcopy(BASE_KJOB_TEMPLATE)
-
-  decorated_manifest = tcpx_decorator.decorate_kjob_template(kjob_template)
-
-  pod_template_spec = decorated_manifest["spec"]["template"]["spec"]
-
-  # Check annotations are NOT added
-  assert "annotations" not in decorated_manifest["spec"]["template"].get(
-      "metadata", {}
-  )
-
-  # Check tolerations
-  tolerations = pod_template_spec["tolerations"]
-  assert {
-      "key": "user-workload",
-      "operator": "Equal",
-      "value": "true",
-      "effect": "NoSchedule",
-  } in tolerations
-
-  # Check volumes
-  volumes = pod_template_spec["volumes"]
-  volume_names = {v["name"] for v in volumes}
-  assert "libraries" in volume_names
-  assert "sys" in volume_names
-  assert "proc-sys" in volume_names
-  assert "tcpx-socket" in volume_names
-  assert "dshm" in volume_names
-
-  # Check init container
-  init_containers = pod_template_spec["initContainers"]
-  assert len(init_containers) == 1
-  tcpx_daemon = init_containers[0]
-  assert tcpx_daemon["name"] == "tcpx-daemon"
-  assert tcpx_daemon["image"].endswith(f":{tcpx_decorator.tcpx}")
-
-  # Check GPU container update
-  gpu_container = pod_template_spec["containers"][0]
-  assert gpu_container["name"] == "main-gpu-container"
-
-  # Check env
-  env_vars = {e["name"]: e["value"] for e in gpu_container["env"]}
-  assert env_vars["LD_LIBRARY_PATH"] == "/usr/local/nvidia/lib64"
-
-  # Check volume mounts
-  volume_mounts = {
-      vm["name"]: vm["mountPath"] for vm in gpu_container["volumeMounts"]
-  }
-  assert volume_mounts["tcpx-socket"] == "/tmp"
-  assert volume_mounts["libraries"] == "/usr/local/nvidia/lib64"
-  assert volume_mounts["dshm"] == "/dev/shm"
-
-  # Check non-GPU container is not updated
-  sidecar_container = pod_template_spec["containers"][1]
-  assert "env" not in sidecar_container
-  assert "volumeMounts" not in sidecar_container

xpk/core/workload_decorators/tcpxo_decorator.py

@@ -22,22 +22,6 @@ from ...utils.yaml import literal_string
 rxdm = 'v1.0.12'
 
 
-def decorate_kjob_template(job_manifest: dict) -> dict:
-  spec = (
-      job_manifest.setdefault('spec', {})
-      .setdefault('template', {})
-      .setdefault('spec', {})
-  )
-  spec.setdefault('tolerations', [])
-  spec.setdefault('volumes', [])
-
-  add_volumes(job_manifest)
-  add_tolerations(job_manifest)
-  add_tcpxo_daemon_container(job_manifest)
-  update_gpu_containers(job_manifest)
-  return job_manifest
-
-
 def decorate_job(job_manifest: dict, sub_networks: list[str]) -> dict:
   job_manifest.setdefault('spec', {}).setdefault('template', {}).setdefault(
       'metadata', {}

xpk/parser/common.py

@@ -180,157 +180,6 @@ def add_global_arguments(custom_parser_or_group: ParserOrArgumentGroup):
   )
 
 
-def add_slurm_arguments(custom_parser_or_group: ParserOrArgumentGroup):
-  """Add Slurm job arguments to the parser.
-
-  Args:
-    custom_parser_or_group: parser or argument group to add global arguments to.
-  """
-  custom_parser_or_group.add_argument(
-      '--ignore-unknown-flags',
-      type=bool,
-      action=argparse.BooleanOptionalAction,
-      default=False,
-      help='Ignore all the unsupported flags in the bash script.',
-  )
-  custom_parser_or_group.add_argument(
-      '-a',
-      '--array',
-      type=str,
-      default=None,
-      help=(
-          'Submit a job array, multiple jobs to be executed with identical'
-          ' parameters. The indexes specification identifies what array index'
-          ' values should be used. For example, "--array=0-15" or'
-          ' "--array=0,6,16-32". Multiple values may be specified using a comma'
-          ' separated list and/or a range of values with a "-" separator. For'
-          ' example "--array=0-15%%4" will limit the number of simultaneously'
-          ' running tasks from this job array to 4. The minimum index value is'
-          ' 0. The maximum index value is 2147483647.'
-      ),
-  )
-  custom_parser_or_group.add_argument(
-      '-c',
-      '--cpus-per-task',
-      type=str,
-      default=None,
-      help='How much cpus a container inside a pod requires.',
-  )
-  custom_parser_or_group.add_argument(
-      '--gpus-per-task',
-      type=str,
-      default=None,
-      help='How much gpus a container inside a pod requires.',
-  )
-  custom_parser_or_group.add_argument(
-      '--mem',
-      type=str,
-      default=None,
-      help='How much memory a pod requires.',
-  )
-  custom_parser_or_group.add_argument(
-      '--mem-per-task',
-      type=str,
-      default=None,
-      help='How much memory a container requires.',
-  )
-  custom_parser_or_group.add_argument(
-      '--mem-per-cpu',
-      type=str,
-      default=None,
-      help=(
-          'How much memory a container requires, it multiplies the number '
-          'of requested cpus per task by mem-per-cpu.'
-      ),
-  )
-  custom_parser_or_group.add_argument(
-      '--mem-per-gpu',
-      type=str,
-      default=None,
-      help=(
-          'How much memory a container requires, it multiplies the number '
-          'of requested gpus per task by mem-per-gpu.'
-      ),
-  )
-  custom_parser_or_group.add_argument(
-      '-N',
-      '--nodes',
-      type=int,
-      default=None,
-      help='Number of pods to be used at a time.',
-  )
-  custom_parser_or_group.add_argument(
-      '-n',
-      '--ntasks',
-      type=int,
-      default=None,
-      help='Number of identical containers inside of a pod, usually 1.',
-  )
-  custom_parser_or_group.add_argument(
-      '-o',
-      '--output',
-      type=str,
-      default=None,
-      help=(
-          'Where to redirect the standard output stream of a task. If not'
-          ' passed it proceeds to stdout, and is available via kubectl logs.'
-      ),
-  )
-  custom_parser_or_group.add_argument(
-      '-e',
-      '--error',
-      type=str,
-      default=None,
-      help=(
-          'Where to redirect std error stream of a task. If not passed it'
-          ' proceeds to stdout, and is available via kubectl logs.'
-      ),
-  )
-  custom_parser_or_group.add_argument(
-      '--input',
-      type=str,
-      default=None,
-      help='What to pipe into the script.',
-  )
-  custom_parser_or_group.add_argument(
-      '-J',
-      '--job-name',
-      type=str,
-      default=None,
-      help='What is the job name.',
-  )
-  custom_parser_or_group.add_argument(
-      '-D',
-      '--chdir',
-      type=str,
-      default=None,
-      help='Change directory before executing the script.',
-  )
-  custom_parser_or_group.add_argument(
-      '-t',
-      '--time',
-      type=str,
-      default=None,
-      help=(
-          'Set a limit on the total run time of the job. '
-          'A time limit of zero requests that no time limit be imposed. '
-          'Acceptable time formats include "minutes", "minutes:seconds", '
-          '"hours:minutes:seconds", "days-hours", "days-hours:minutes" '
-          'and "days-hours:minutes:seconds".'
-      ),
-  )
-  custom_parser_or_group.add_argument(
-      '--priority',
-      type=str,
-      default='medium',
-      choices=['very-low', 'low', 'medium', 'high', 'very-high'],
-      help=(
-          'A priority, one of `very-low`, `low`, `medium`, `high` or'
-          ' `very-high`. Defaults to `medium`.'
-      ),
-  )
-
-
 def add_tpu_type_argument(
     custom_parser_or_group: ParserOrArgumentGroup,
     required: bool = False,