def Args(parser):
flags.AddRegionResourceArg(parser, 'to create index')
flags.GetDisplayNameArg('index').AddToParser(parser)
flags.GetDescriptionArg('index').AddToParser(parser)
flags.GetMetadataFilePathArg('index',
required=True).AddToParser(parser)
flags.GetMetadataSchemaUriArg('index').AddToParser(parser)
labels_util.AddCreateLabelsFlags(parser)
def _AddArgs(parser):
flags.GetDisplayNameArg('endpoint').AddToParser(parser)
flags.AddRegionResourceArg(parser,
'to create endpoint',
prompt_func=region_util.PromptForOpRegion)
flags.GetDescriptionArg('endpoint').AddToParser(parser)
flags.GetUserSpecifiedIdArg('endpoint').AddToParser(parser)
labels_util.AddCreateLabelsFlags(parser)
flags.GetEndpointNetworkArg().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 _Args(parser):
"""Add flags for create command."""
flags.AddRegionResourceArg(parser,
'to create model deployment monitoring job')
flags.GetDisplayNameArg('model deployment monitoring job').AddToParser(
parser)
flags.GetEndpointIdArg(required=True).AddToParser(parser)
flags.GetEmailsArg(required=True).AddToParser(parser)
flags.GetPredictionSamplingRateArg(required=True).AddToParser(parser)
flags.GetMonitoringFrequencyArg(required=False).AddToParser(parser)
flags.GetPredictInstanceSchemaArg(required=False).AddToParser(parser)
flags.GetAnalysisInstanceSchemaArg(required=False).AddToParser(parser)
flags.GetSamplingPredictRequestArg(required=False).AddToParser(parser)
flags.GetMonitoringLogTtlArg(required=False).AddToParser(parser)
flags.AddObjectiveConfigGroupForCreate(parser, required=False)
flags.AddKmsKeyResourceArg(parser, 'model deployment monitoring job')
flags.GetAnomalyCloudLoggingArg(required=False).AddToParser(parser)
labels_util.AddCreateLabelsFlags(parser)
def AddCreateCustomJobFlags(parser, version=constants.BETA_VERSION):
"""Adds flags related to create a custom job."""
shared_flags.AddRegionResourceArg(parser, 'to create a custom job')
shared_flags.TRAINING_SERVICE_ACCOUNT.AddToParser(parser)
shared_flags.NETWORK.AddToParser(parser)
shared_flags.AddKmsKeyResourceArg(parser, 'custom job')
_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)
if version == constants.GA_VERSION:
worker_pool_spec_group.AddArgument(_WORKER_POOL_SPEC_GA)
else:
worker_pool_spec_group.AddArgument(_WORKER_POOL_SPEC_BETA)
worker_pool_spec_group.AddToParser(parser)
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)
def Args(parser):
flags.AddRegionResourceArg(parser,
'to list model deployment monitoring jobs')
def _AddArgs(parser):
flags.AddRegionResourceArg(parser, 'to create a Tensorboard')
flags.GetDisplayNameArg('tensorboard').AddToParser(parser)
flags.GetDescriptionArg('tensorboard').AddToParser(parser)
flags.AddKmsKeyResourceArg(parser, 'tensorboard')
labels_util.AddCreateLabelsFlags(parser)
def Args(parser):
parser.display_info.AddFormat(_DEFAULT_FORMAT)
parser.display_info.AddUriFunc(_GetUri)
flags.AddRegionResourceArg(parser, 'to list endpoints')
def Args(parser):
flags.AddRegionResourceArg(parser, 'to list index endpoints')
def Args(parser):
flags.AddRegionResourceArg(parser, 'to create index endpoint')
flags.GetDisplayNameArg('index endpoint').AddToParser(parser)
flags.GetDescriptionArg('index endpoint').AddToParser(parser)
flags.GetNetworkArg().AddToParser(parser)
labels_util.AddCreateLabelsFlags(parser)
def Args(parser):
flags.AddRegionResourceArg(parser, 'to list models')
def _AddArgs(parser):
parser.display_info.AddFormat(_DEFAULT_FORMAT)
parser.display_info.AddUriFunc(_GetUri)
flags.AddRegionResourceArg(parser,
'to list endpoints',
prompt_func=region_util.PromptForOpRegion)
def Args(parser):
flags.AddRegionResourceArg(parser, 'to list custom jobs')
def Args(parser):
parser.display_info.AddFormat(_DEFAULT_FORMAT)
flags.AddRegionResourceArg(parser,
'to list models',
prompt_func=region_util.PromptForOpRegion)
def Args(parser):
flags.AddRegionResourceArg(parser, 'to list Tensorboards')
parser.display_info.AddUriFunc(_GetUriGa)
def Args(parser):
flags.AddRegionResourceArg(parser,
'to list hyperparameter tuning jobs')
