Adding the ssd flag to change the disk type (#1211)

GoogleCloudDataproc · Aug 8, 2024 · d4d557a · d4d557a
1 parent 51690b6
commit d4d557a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/integration_tests/dataproc_test_case.py b/integration_tests/dataproc_test_case.py
@@ -165,6 +165,9 @@ def createCluster(self,
         args.append("--master-boot-disk-size={}".format(boot_disk_size))
         args.append("--worker-boot-disk-size={}".format(boot_disk_size))
 
+        args.append("--master-boot-disk-type=pd-ssd")
+        args.append("--worker-boot-disk-type=pd-ssd")
+
         args.append("--format=json")
 
         args.append("--region={}".format(self.REGION))