Skip to content

Commit 3c76270

Browse files
SinaChavoshiTensorflow Cloud maintainers
authored andcommitted
Change default Cloud Build instance to N1_HIGHCPU_8.
PiperOrigin-RevId: 356554444
1 parent f890397 commit 3c76270

File tree

3 files changed

+13
-0
lines changed

3 files changed

+13
-0
lines changed

RELEASE.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
# Current Version (Still in Development)
22

33
* Add notes for next release here.
4+
* Upgrad the default Cloud Build machine to `N1_HIGHCPU_8`
45

56
# Release 0.1.12
67

src/python/tensorflow_cloud/core/containerize.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -562,4 +562,7 @@ def _create_cloud_build_request_dict(
562562
"object": storage_object_name,
563563
}
564564
}
565+
request_dict["options"] = {
566+
"machineType": "N1_HIGHCPU_8"
567+
}
565568
return request_dict

src/python/tensorflow_cloud/core/tests/unit/containerize_test.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -446,6 +446,9 @@ def test_get_docker_image_cloud_build(
446446
request_dict["projectId"] = self.project_id
447447
request_dict["timeout"] = "1200s"
448448
request_dict["images"] = [["gcr.io/my-project/tf_cloud_train:abcde"]]
449+
request_dict["options"] = {
450+
"machineType": "N1_HIGHCPU_8"
451+
}
449452
request_dict["steps"] = [{
450453
"name": "gcr.io/cloud-builders/docker",
451454
"args": [
@@ -615,6 +618,9 @@ def test_get_docker_image_cloud_build_with_custom_image_uri(
615618
request_dict["projectId"] = self.project_id
616619
request_dict["timeout"] = "1200s"
617620
request_dict["images"] = [[img_tag]]
621+
request_dict["options"] = {
622+
"machineType": "N1_HIGHCPU_8"
623+
}
618624
request_dict["steps"] = [{
619625
"name": "gcr.io/cloud-builders/docker",
620626
"entrypoint": "bash",
@@ -736,6 +742,9 @@ def test_get_docker_image_cloud_build_with_cache(
736742
request_dict["projectId"] = self.project_id
737743
request_dict["timeout"] = "1200s"
738744
request_dict["images"] = [[img_tag]]
745+
request_dict["options"] = {
746+
"machineType": "N1_HIGHCPU_8"
747+
}
739748
request_dict["steps"] = [{
740749
"name": "gcr.io/cloud-builders/docker",
741750
"entrypoint": "bash",

0 commit comments

Comments
 (0)