commit 17a15df7e4512a2aaa9f02e1ceb0df27861d4db4
author Austin Anderson <angerson@google.com> Wed Aug 12 10:51:02 2020 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Wed Aug 12 11:00:27 2020 -0700
tree d4ef616d087b122c061590ad9dabc479e5ace72f
parent 42b2ba9b712750782f82ea1626a569c60661b6b3

Add Kokoro build ID for more analysis

PiperOrigin-RevId: 326265653
Change-Id: I0f7ce0d1ce58c380389e969a4604e93e7f4fd90e

tensorflow/tools/ci_build/sizetrack_helper.py

diff --git a/tensorflow/tools/ci_build/sizetrack_helper.py b/tensorflow/tools/ci_build/sizetrack_helper.py
index 03a04e3..e56009d 100755
--- a/tensorflow/tools/ci_build/sizetrack_helper.py
+++ b/tensorflow/tools/ci_build/sizetrack_helper.py
@@ -100,6 +100,10 @@
     type=str,
     help="Name of job calling this script. Default: $KOKORO_JOB_NAME.")
 parser.add_argument(
+    "--build_id",
+    type=str,
+    help="UUID of build calling this script. Default: $KOKORO_BUILD_ID.")
+parser.add_argument(
     "--print_schema",
     action="store_true",
     help="Print the table schema and don't do anything else.")
@@ -151,6 +155,7 @@
     "logged_date:timestamp",
     "uploaded_to:string",
     "job:string",
+    "build_id:string",
 ])
 # Select the earliest recorded commit in the same table for the same artifact
 # and team. Used to determine the full range of tested commits for each
@@ -328,6 +333,7 @@
       current_time,
       get_upload_path(),
       FLAGS.job,
+      FLAGS.build_id,
   ]
 
 
@@ -347,6 +353,8 @@
 
   if not FLAGS.job:
     FLAGS.job = os.environ.get("KOKORO_JOB_NAME", "NO_JOB")
+  if not FLAGS.build_id:
+    FLAGS.build_id = os.environ.get("KOKORO_BUILD_ID", "NO_BUILD")
 
   # Generate data about this artifact into a Tab Separated Value file
   next_tsv_row = build_row()