From 334366546501833149479556b55bfbc3c9562236 Mon Sep 17 00:00:00 2001
From: Brad Miro <bmiro@google.com>
Date: Fri, 26 Mar 2021 01:54:03 -0400
Subject: [PATCH] fix: (samples) fixing samples for new machine types (#150)

Dataproc 2.0 is the new default image which does not support n1-standard-1.

Fixes https://github.com/googleapis/python-dataproc/issues/147, https://github.com/googleapis/python-dataproc/issues/148, https://github.com/googleapis/python-dataproc/issues/149
---
 samples/snippets/create_cluster.py        | 4 ++--
 samples/snippets/quickstart/quickstart.py | 4 ++--
 samples/snippets/submit_job_test.py       | 4 ++--
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/samples/snippets/create_cluster.py b/samples/snippets/create_cluster.py
index 732b349a..f4fee7d8 100644
--- a/samples/snippets/create_cluster.py
+++ b/samples/snippets/create_cluster.py
@@ -47,8 +47,8 @@ def create_cluster(project_id, region, cluster_name):
         "project_id": project_id,
         "cluster_name": cluster_name,
         "config": {
-            "master_config": {"num_instances": 1, "machine_type_uri": "n1-standard-1"},
-            "worker_config": {"num_instances": 2, "machine_type_uri": "n1-standard-1"},
+            "master_config": {"num_instances": 1, "machine_type_uri": "n1-standard-2"},
+            "worker_config": {"num_instances": 2, "machine_type_uri": "n1-standard-2"},
         },
     }
 
diff --git a/samples/snippets/quickstart/quickstart.py b/samples/snippets/quickstart/quickstart.py
index 45e7f348..c9e73002 100644
--- a/samples/snippets/quickstart/quickstart.py
+++ b/samples/snippets/quickstart/quickstart.py
@@ -44,8 +44,8 @@ def quickstart(project_id, region, cluster_name, job_file_path):
         "project_id": project_id,
         "cluster_name": cluster_name,
         "config": {
-            "master_config": {"num_instances": 1, "machine_type_uri": "n1-standard-1"},
-            "worker_config": {"num_instances": 2, "machine_type_uri": "n1-standard-1"},
+            "master_config": {"num_instances": 1, "machine_type_uri": "n1-standard-2"},
+            "worker_config": {"num_instances": 2, "machine_type_uri": "n1-standard-2"},
         },
     }
 
diff --git a/samples/snippets/submit_job_test.py b/samples/snippets/submit_job_test.py
index 0cad6248..326b38d5 100644
--- a/samples/snippets/submit_job_test.py
+++ b/samples/snippets/submit_job_test.py
@@ -30,11 +30,11 @@
     'config': {
         'master_config': {
             'num_instances': 1,
-            'machine_type_uri': 'n1-standard-1'
+            'machine_type_uri': 'n1-standard-2'
         },
         'worker_config': {
             'num_instances': 2,
-            'machine_type_uri': 'n1-standard-1'
+            'machine_type_uri': 'n1-standard-2'
         }
     }
 }