def Run(self, args): """This is what gets called when the user runs this command. Args: args: an argparse namespace. All the arguments that were provided to this command invocation. Returns: Some value that we want to have printed later. """ client = models.ModelsClient() op = client.Delete(args.model) return operations.OperationsClient().WaitForOperation( op, message='Deleting model [{}]'.format(args.model)).response
def WaitForOpMaybe(op, async_=False, message=None): """Waits for an operation if async_ flag is on. Args: op: Cloud ML operation, the operation to poll async_: bool, whether to wait for the operation or return immediately message: str, the message to display while waiting for the operation Returns: The result of the operation if async_ is true, or the Operation message otherwise """ if async_: return op return operations.OperationsClient().WaitForOperation( op, message=message).response
def Run(self, args): return operations.OperationsClient().List( properties.VALUES.core.project.Get())
def Run(self, args): operation_ref = resources.REGISTRY.Parse( args.operation, collection='ml.projects.operations') return operations.OperationsClient().Get(operation_ref)
def Run(self, args): return operations_util.Wait(operations.OperationsClient('v1'), args.operation)
def Run(self, args): models_client = models.ModelsClient('v1beta1') operations_client = operations.OperationsClient('v1beta1') return models_util.Delete(models_client, operations_client, args.model)
def Run(self, args): project_ref = resources.REGISTRY.Parse( properties.VALUES.core.project.Get(required=True), collection='ml.projects') return operations.OperationsClient().List(project_ref)
def Run(self, args): return operations_util.Describe(operations.OperationsClient('v1'), args.operation)
def Run(self, args): return operations_util.Delete(operations.OperationsClient('v1beta1'), args.operation)
def Run(self, args): return operations_util.List(operations.OperationsClient('v1'))
def Run(self, args): operations_client = operations.OperationsClient() operation_ref = resources.REGISTRY.Parse( args.operation, collection='ml.projects.operations') operation = operations_client.Get(operation_ref) return operations_client.WaitForOperation(operation)
def Run(self, args): return versions_util.Delete(versions_api.VersionsClient('v1'), operations.OperationsClient('v1'), args.version, model=args.model)