Ajout de paramètres de type machine en fonction Google Cloud SDK Python create_cluster()

Question

python de Google Cloud docs ont un script (python-docs-échantillons/Dataproc/submit_job_to_cluster.py) qui a la fonction suivante:

def create_cluster(dataproc, project, zone, region, cluster_name): 
     print('Creating cluster...') 
     zone_uri = 'https://www.googleapis.com/compute/v1/projects/{}/zones/{}'.format(
     project, zone) 
     cluster_data = { 
     'projectId': project, 
     'clusterName': cluster_name, 
     'config': { 
      'gceClusterConfig': { 
       'zoneUri': zone_uri 
      } 
     } 
     } 
     result = dataproc.projects().regions().clusters().create(
     projectId=project, 
     region=region, 
     body=cluster_data).execute() 
     return result 

Je me demandais s'il est possible de spécifier les types de machine pour les nœuds maître et travailleur du cluster dans cette fonction?

Answer 1

Ce qui suit devrait fonctionner:

def create_cluster(dataproc, project, zone, region, cluster_name): 
     print('Creating cluster...') 
     zone_uri = 'https://www.googleapis.com/compute/v1/projects/{}/zones/{}'.format(
     project, zone) 
     cluster_data = { 
     'projectId': project, 
     'clusterName': cluster_name, 
     'config': { 
      'gceClusterConfig': { 
       'zoneUri': zone_uri 
       }, 
       'masterConfig': { 
       'machineTypeUri' : 'n1-standard-1', 
       }, 
       'workerConfig': { 
       'machineTypeUri' : 'n1-standard-4', 
       }, 
      } 
     } 
     } 
     result = dataproc.projects().regions().clusters().create(
     projectId=project, 
     region=region, 
     body=cluster_data).execute() 
     return result 

https://cloud.google.com/dataproc/docs/reference/rest/v1/projects.regions.clusters#ClusterConfig