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)