(context)
return {'resources': resources}
GenerateConfig returns the deployment manager config.
GenerateConfig returns the deployment manager config.
GenerateConfig returns the deployment manager config .
def GenerateConfig(context):
  """GenerateConfig returns the deployment manager config."""
  project = context.env['project']
  zone = context.properties['zone']
  tpu_name = 'tpu-%s' % context.env['deployment']
  accelerator_type = (
      context.properties['acceleratorType']
      if 'acceleratorType' in context.properties else 'v2-8')
  tensorflow_version = (
      context.properties['tensorflowVersion']
      if 'tensorflowVersion' in context.properties else '2.3')
  cidr_block = context.properties['cidrBlock']
  resources = []
  resources.append({
      'name': tpu_name,
      'type': 'gcp-types/tpu-v1alpha1:projects.locations.nodes',
      'properties': {
          'nodeId': tpu_name,
          'parent': 'projects/%s/locations/%s' % (project, zone),
          'acceleratorType': accelerator_type,
          'network': context.properties.get('network') or 'default',
          'cidrBlock': cidr_block,
          'tensorflowVersion': tensorflow_version,
      }
  })
  gcs_bucket = context.properties.get('gcsBucket')
  if gcs_bucket:
    resources.append({
        'name': gcs_bucket,
        'type': 'gcp-types/storage-v1:buckets',
        'properties': {
            'location': 'US',
            'storageClass': 'STANDARD',
        }
    })

    resources.append({
        'name': 'tpu-bucket-iam',
        'type': 'gcp-types/storage-v1:virtual.buckets.iamMemberBinding',
        'properties': {
            'bucket': '$(ref.%s.name)' % gcs_bucket,
            'member': 'serviceAccount:$(ref.%s.serviceAccount)' % tpu_name,
            'role': 'roles/storage.admin'
        }
    })

  return {'resources': resources}
def GenerateConfig ( context ) : project = context . env [ 'project' ] zone = context . properties [ 'zone' ] tpu_name = 'tpu-%s' % context . env [ 'deployment' ] accelerator_type = ( context . properties [ 'acceleratorType' ] if 'acceleratorType' in context . properties else 'v2-8' ) tensorflow_version = ( context . properties [ 'tensorflowVersion' ] if 'tensorflowVersion' in context . properties else '2.3' ) cidr_block = context . properties [ 'cidrBlock' ] resources = [ ] resources . append ( { 'name' : tpu_name , 'type' : 'gcp-types/tpu-v1alpha1:projects.locations.nodes' , 'properties' : { 'nodeId' : tpu_name , 'parent' : 'projects/%s/locations/%s' % ( project , zone ) , 'acceleratorType' : accelerator_type , 'network' : context . properties . get ( 'network' ) or 'default' , 'cidrBlock' : cidr_block , 'tensorflowVersion' : tensorflow_version , } } ) gcs_bucket = context . properties . get ( 'gcsBucket' ) if gcs_bucket : resources . append ( { 'name' : gcs_bucket , 'type' : 'gcp-types/storage-v1:buckets' , 'properties' : { 'location' : 'US' , 'storageClass' : 'STANDARD' , } } ) resources . append ( { 'name' : 'tpu-bucket-iam' , 'type' : 'gcp-types/storage-v1:virtual.buckets.iamMemberBinding' , 'properties' : { 'bucket' : '$(ref.%s.name)' % gcs_bucket , 'member' : 'serviceAccount:$(ref.%s.serviceAccount)' % tpu_name , 'role' : 'roles/storage.admin' } } ) return { 'resources' : resources }