def _GetModelDeploymentResourceType(model_ref, client): """Gets the deployment resource type of a model. Each model resource belongs to exactly one supported deployment resource type. The value is the first item in the list, the length of which must be one. Args: model_ref: a model resource object. client: an apis.GetClientInstance object. Returns: A string which value must be 'DEDICATED_RESOURCES' or 'AUTOMATIC_RESOURCES'. Raises: ArgumentError: if the model resource object is not found. """ try: model_msg = model_client.ModelsClient(client=client).Get(model_ref) except apitools_exceptions.HttpError: raise errors.ArgumentError( ('There is an error while getting the model information. ' 'Please make sure the model %r exists.' % model_ref.RelativeName())) model_resource = encoding.MessageToPyValue(model_msg) return model_resource['supportedDeploymentResourcesTypes'][0]
def Run(self, args): region_ref = args.CONCEPTS.region.Parse() region = region_ref.AsDict()['locationsId'] with endpoint_util.AiplatformEndpointOverrides( version=constants.GA_VERSION, region=region): client_instance = apis.GetClientInstance( constants.AI_PLATFORM_API_NAME, constants.AI_PLATFORM_API_VERSION[constants.GA_VERSION]) operation = client.ModelsClient( client=client_instance, messages=client_instance.MESSAGES_MODULE).UploadV1( region_ref, args.display_name, args.description, args.artifact_uri, args.container_image_uri, args.container_command, args.container_args, args.container_env_vars, args.container_ports, args.container_predict_route, args.container_health_route, explanation_spec=self._BuildExplanationSpec(args)) return operations_util.WaitForOpMaybe( operations_client=operations.OperationsClient( client=client_instance, messages=client_instance.MESSAGES_MODULE), op=operation, op_ref=models_util.ParseModelOperation(operation.name))
def _Run(self, args): model_ref = args.CONCEPTS.model.Parse() region = model_ref.AsDict()['locationsId'] with endpoint_util.AiplatformEndpointOverrides( version=constants.BETA_VERSION, region=region): response = client.ModelsClient().Get(model_ref) return response
def __init__(self, *args, **kwargs): super(UploadV1, self).__init__(*args, **kwargs) client_instance = apis.GetClientInstance( constants.AI_PLATFORM_API_NAME, constants.AI_PLATFORM_API_VERSION[constants.GA_VERSION]) self.messages = client.ModelsClient( client=client_instance, messages=client_instance.MESSAGES_MODULE).messages
def _Run(self, args, model_ref, region): with endpoint_util.AiplatformEndpointOverrides( version=constants.BETA_VERSION, region=region): operation = client.ModelsClient().Delete(model_ref) return operations_util.WaitForOpMaybe( operations_client=operations.OperationsClient(), op=operation, op_ref=models_util.ParseModelOperation(operation.name))
def _Run(self, args, model_ref, region): with endpoint_util.AiplatformEndpointOverrides( version=constants.GA_VERSION, region=region): client_instance = apis.GetClientInstance( constants.AI_PLATFORM_API_NAME, constants.AI_PLATFORM_API_VERSION[constants.GA_VERSION]) return client.ModelsClient( client=client_instance, messages=client_instance.MESSAGES_MODULE).Delete(model_ref)
def _Run(self, args): model_ref = args.CONCEPTS.model.Parse() region = model_ref.AsDict()['locationsId'] with endpoint_util.AiplatformEndpointOverrides( version=constants.BETA_VERSION, region=region): operation = client.ModelsClient().Delete(model_ref) return operations_util.WaitForOpMaybe( operations_client=operations.OperationsClient(), op=operation, op_ref=models_util.ParseModelOperation(operation.name))
def Run(self, args): region_ref = args.CONCEPTS.region.Parse() region = region_ref.AsDict()['locationsId'] with endpoint_util.AiplatformEndpointOverrides( version=constants.BETA_VERSION, region=region): operation = client.ModelsClient().UploadV1Beta1( region_ref, args.display_name, args.description, args.artifact_uri, args.container_image_uri, args.container_command, args.container_args, args.container_env_vars, args.container_ports, args.container_predict_route, args.container_health_route) return operations_util.WaitForOpMaybe( operations_client=operations.OperationsClient(), op=operation, op_ref=models_util.ParseModelOperation(operation.name))
def __init__(self, *args, **kwargs): super(UploadV1Beta1, self).__init__(*args, **kwargs) self.messages = client.ModelsClient().messages
def _Run(self, args, region_ref, region): with endpoint_util.AiplatformEndpointOverrides( version=constants.BETA_VERSION, region=region): return client.ModelsClient().List(region_ref=region_ref)
def _Run(self, args): region_ref = args.CONCEPTS.region.Parse() region = region_ref.AsDict()['locationsId'] with endpoint_util.AiplatformEndpointOverrides( version=constants.BETA_VERSION, region=region): return client.ModelsClient().List(region_ref=region_ref)
def _Run(self, args, model_ref, region): with endpoint_util.AiplatformEndpointOverrides( version=constants.BETA_VERSION, region=region): response = client.ModelsClient().Get(model_ref) return response