Skip to content

Commit

Permalink
Pin tensorflow_version on TPU Node Full Test so that the test passes. F…
Browse files Browse the repository at this point in the history
  • Loading branch information
zeleena authored and Yuan Chuan Kee committed Jan 21, 2024
1 parent 02b036c commit d6ecc42
Showing 1 changed file with 7 additions and 4 deletions.
11 changes: 7 additions & 4 deletions mmv1/templates/terraform/examples/tpu_node_full_test.tf.erb
Original file line number Diff line number Diff line change
@@ -1,6 +1,3 @@
data "google_tpu_tensorflow_versions" "available" {
}

<%#-
WARNING: cidr_block must not overlap with other existing TPU blocks
Make sure if you change this value that it does not overlap with the
Expand All @@ -13,7 +10,13 @@ resource "google_tpu_node" "<%= ctx[:primary_resource_id] %>" {

accelerator_type = "v3-8"

tensorflow_version = data.google_tpu_tensorflow_versions.available.versions[0]
<%#-
We previously used the first available version from the
google_tpu_tensorflow_versions data source. However, this started to return a
random set of versions which caused our tests to occasionally fail, so we pin
tensorflow_version to a specific version so that our tests pass reliably.
-%>
tensorflow_version = "2.10.0"

description = "Terraform Google Provider test TPU"
use_service_networking = true
Expand Down

0 comments on commit d6ecc42

Please sign in to comment.