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.GetModuleNameFlag().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)