From 689e077cd5fd0bec584776089aa3c80017270e19 Mon Sep 17 00:00:00 2001
From: Na Li <linazhao@google.com>
Date: Thu, 26 Mar 2020 11:41:23 -0700
Subject: [PATCH] Fix nightly build.

---
 cloudbuild.yml           | 7 +++++++
 tfjs-core/cloudbuild.yml | 3 +--
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/cloudbuild.yml b/cloudbuild.yml
index 30c0ac949cc..fb2c9cd1144 100644
--- a/cloudbuild.yml
+++ b/cloudbuild.yml
@@ -30,6 +30,7 @@ steps:
   id: 'tfjs-converter'
   args: ['./scripts/run-build.sh', 'tfjs-converter']
   waitFor: ['find-affected-packages']
+  env: ['NIGHTLY=$_NIGHTLY']
 
 # Data.
 - name: 'gcr.io/cloud-builders/gcloud'
@@ -37,6 +38,7 @@ steps:
   id: 'tfjs-data'
   args: ['./scripts/run-build.sh', 'tfjs-data']
   waitFor: ['find-affected-packages']
+  env: ['NIGHTLY=$_NIGHTLY']
 
 # Layers.
 - name: 'gcr.io/cloud-builders/gcloud'
@@ -44,6 +46,7 @@ steps:
   id: 'tfjs-layers'
   args: ['./scripts/run-build.sh', 'tfjs-layers']
   waitFor: ['find-affected-packages']
+  env: ['NIGHTLY=$_NIGHTLY']
 
 # Union.
 - name: 'gcr.io/cloud-builders/gcloud'
@@ -51,6 +54,7 @@ steps:
   id: 'tfjs'
   args: ['./scripts/run-build.sh', 'tfjs']
   waitFor: ['find-affected-packages']
+  env: ['NIGHTLY=$_NIGHTLY']
 
 # Vis.
 - name: 'gcr.io/cloud-builders/gcloud'
@@ -72,6 +76,7 @@ steps:
   id: 'tfjs-backend-wasm'
   args: ['./scripts/run-build.sh', 'tfjs-backend-wasm']
   waitFor: ['find-affected-packages']
+  env: ['NIGHTLY=$_NIGHTLY']
 
 # React Native.
 - name: 'gcr.io/cloud-builders/gcloud'
@@ -86,6 +91,7 @@ steps:
   id: 'tfjs-node'
   args: ['./scripts/run-build.sh', 'tfjs-node']
   waitFor: ['find-affected-packages']
+  env: ['NIGHTLY=$_NIGHTLY']
 
 # Node GPU.
 - name: 'gcr.io/cloud-builders/gcloud'
@@ -93,6 +99,7 @@ steps:
   id: 'tfjs-node-gpu'
   args: ['./scripts/run-build.sh', 'tfjs-node-gpu']
   waitFor: ['find-affected-packages']
+  env: ['NIGHTLY=$_NIGHTLY']
 
 # Integration tests
 - name: 'node:10'
diff --git a/tfjs-core/cloudbuild.yml b/tfjs-core/cloudbuild.yml
index 28d69c0b2ce..6a88c268139 100644
--- a/tfjs-core/cloudbuild.yml
+++ b/tfjs-core/cloudbuild.yml
@@ -28,9 +28,8 @@ steps:
   entrypoint: 'yarn'
   args: ['test-ci']
   waitFor: ['build']
-  env: ['BROWSERSTACK_USERNAME=deeplearnjs1']
+  env: ['BROWSERSTACK_USERNAME=deeplearnjs1', 'NIGHTLY=$_NIGHTLY']
   secretEnv: ['BROWSERSTACK_KEY']
-  env: ['NIGHTLY=$_NIGHTLY']
 
 # Run unit tests in react native.
 - name: 'node:10'