def AddCustomJobResourceArg(parser,
verb,
regions=constants.SUPPORTED_TRAINING_REGIONS):
"""Add a resource argument for a Vertex AI custom job.
NOTE: Must be used only if it's the only resource arg in the command.
Args:
parser: the parser for the command.
verb: str, the verb to describe the job resource, such as 'to update'.
regions: list[str], the list of supported regions.
"""
job_resource_spec = concepts.ResourceSpec(
resource_collection=custom_jobs_util.CUSTOM_JOB_COLLECTION,
resource_name='custom job',
locationsId=shared_flags.RegionAttributeConfig(
prompt_func=region_util.GetPromptForRegionFunc(regions)),
projectsId=concepts.DEFAULT_PROJECT_ATTRIBUTE_CONFIG,
disable_auto_completers=False)
concept_parsers.ConceptParser.ForResource(
'custom_job',
job_resource_spec,
'The custom job {}.'.format(verb),
required=True).AddToParser(parser)
def Args(cls, parser):
"""Method called by Calliope to set up arguments for this command.
Args:
parser: A argparse.Parser to register accepted arguments in command input.
"""
flags.AddRegionResourceArg(
parser,
'to list hyperparameter tuning jobs',
prompt_func=region_util.GetPromptForRegionFunc(
constants.SUPPORTED_TRAINING_REGIONS))
flags.AddUriFlags(parser, hp_tuning_jobs_util.HPTUNING_JOB_COLLECTION,
constants.AI_PLATFORM_API_VERSION[cls._version])
def AddCreateHpTuningJobFlags(parser, algorithm_enum):
"""Adds arguments for creating hp tuning job."""
_HPTUNING_JOB_DISPLAY_NAME.AddToParser(parser)
_HPTUNING_JOB_CONFIG.AddToParser(parser)
_HPTUNING_MAX_TRIAL_COUNT.AddToParser(parser)
_HPTUNING_PARALLEL_TRIAL_COUNT.AddToParser(parser)
labels_util.AddCreateLabelsFlags(parser)
flags.AddRegionResourceArg(parser,
'to create a hyperparameter tuning job',
prompt_func=region_util.GetPromptForRegionFunc(
constants.SUPPORTED_TRAINING_REGIONS))
flags.TRAINING_SERVICE_ACCOUNT.AddToParser(parser)
flags.NETWORK.AddToParser(parser)
flags.ENABLE_WEB_ACCESS.AddToParser(parser)
flags.AddKmsKeyResourceArg(parser, 'hyperparameter tuning job')
arg_utils.ChoiceEnumMapper(
'--algorithm',
algorithm_enum,
help_str='Search algorithm specified for the given study. '
).choice_arg.AddToParser(parser)
def AddCreateCustomJobFlags(parser):
"""Adds flags related to create a custom job."""
shared_flags.AddRegionResourceArg(
parser,
'to create a custom job',
prompt_func=region_util.GetPromptForRegionFunc(
constants.SUPPORTED_TRAINING_REGIONS))
shared_flags.TRAINING_SERVICE_ACCOUNT.AddToParser(parser)
shared_flags.NETWORK.AddToParser(parser)
shared_flags.ENABLE_WEB_ACCESS.AddToParser(parser)
shared_flags.AddKmsKeyResourceArg(parser, 'custom job')
labels_util.AddCreateLabelsFlags(parser)
_DISPLAY_NAME.AddToParser(parser)
_PYTHON_PACKAGE_URIS.AddToParser(parser)
_CUSTOM_JOB_ARGS.AddToParser(parser)
_CUSTOM_JOB_COMMAND.AddToParser(parser)
worker_pool_spec_group = base.ArgumentGroup(
help='Worker pool specification.', required=True)
worker_pool_spec_group.AddArgument(_CUSTOM_JOB_CONFIG)
worker_pool_spec_group.AddArgument(_WORKER_POOL_SPEC)
worker_pool_spec_group.AddToParser(parser)