From 4753fa00ee4afd4c1ad3630e0b812df2cf27edae Mon Sep 17 00:00:00 2001
From: Na Li <linazhao@google.com>
Date: Wed, 25 Mar 2020 14:29:23 -0700
Subject: [PATCH] Properly propagate NIGHTLY flag.

---
 scripts/run-build.sh     | 3 ++-
 tfjs-core/cloudbuild.yml | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/scripts/run-build.sh b/scripts/run-build.sh
index 118b9693830..8c15e147117 100755
--- a/scripts/run-build.sh
+++ b/scripts/run-build.sh
@@ -18,5 +18,6 @@ set -e
 
 DIR=$1
 if [[ -f "$DIR/run-ci" || "$NIGHTLY" = true ]]; then
-  gcloud builds submit . --config=$DIR/cloudbuild.yml
+  gcloud builds submit . --config=$DIR/cloudbuild.yml \
+    --substitutions _NIGHTLY=$NIGHTLY
 fi
diff --git a/tfjs-core/cloudbuild.yml b/tfjs-core/cloudbuild.yml
index 5872e71a53e..28d69c0b2ce 100644
--- a/tfjs-core/cloudbuild.yml
+++ b/tfjs-core/cloudbuild.yml
@@ -30,6 +30,7 @@ steps:
   waitFor: ['build']
   env: ['BROWSERSTACK_USERNAME=deeplearnjs1']
   secretEnv: ['BROWSERSTACK_KEY']
+  env: ['NIGHTLY=$_NIGHTLY']
 
 # Run unit tests in react native.
 - name: 'node:10'