def _AddSubmitTrainingArgs(parser):
"""Add arguments for `jobs submit training` command."""
flags.JOB_NAME.AddToParser(parser)
flags.PACKAGE_PATH.AddToParser(parser)
flags.PACKAGES.AddToParser(parser)
flags.GetModuleNameFlag(required=False).AddToParser(parser)
compute_flags.AddRegionFlag(parser, 'machine learning training job',
'submit')
flags.CONFIG.AddToParser(parser)
flags.STAGING_BUCKET.AddToParser(parser)
flags.GetJobDirFlag(upload_help=True).AddToParser(parser)
flags.GetUserArgs(local=False).AddToParser(parser)
jobs_util.ScaleTierFlagMap().choice_arg.AddToParser(parser)
flags.RUNTIME_VERSION.AddToParser(parser)
flags.AddPythonVersionFlag(parser, 'during training')
sync_group = parser.add_mutually_exclusive_group()
# TODO(b/36195821): Use the flag deprecation machinery when it supports the
# store_true action
sync_group.add_argument(
'--async', action='store_true', help=(
'(DEPRECATED) Display information about the operation in progress '
'without waiting for the operation to complete. '
'Enabled by default and can be omitted; use `--stream-logs` to run '
'synchronously.'))
sync_group.add_argument(
'--stream-logs',
action='store_true',
help=('Block until job completion and stream the logs while the job runs.'
'\n\n'
'Note that even if command execution is halted, the job will still '
'run until cancelled with\n\n'
' $ gcloud ai-platform jobs cancel JOB_ID'))
labels_util.AddCreateLabelsFlags(parser)
def Args(parser): """Register flags for this command.""" flags.PACKAGE_PATH.AddToParser(parser) flags.MODULE_NAME.AddToParser(parser) flags.DISTRIBUTED.AddToParser(parser) flags.PARAM_SERVERS.AddToParser(parser) flags.GetJobDirFlag(upload_help=False, allow_local=True).AddToParser(parser) flags.WORKERS.AddToParser(parser) flags.START_PORT.AddToParser(parser) flags.GetUserArgs(local=True).AddToParser(parser)
