Update from https://crrev.com/329939

build/android/provision_devices.py

Index: build/android/provision_devices.py
diff --git a/build/android/provision_devices.py b/build/android/provision_devices.py
index ee52c7163750e49d65a1759184b39fd75528540a..b366f0c14cdfc6e02f8c00c07ef29aeb998a8dd3 100755
--- a/build/android/provision_devices.py
+++ b/build/android/provision_devices.py
@@ -184,6 +184,12 @@ def SetProperties(device, options):
     except device_errors.CommandFailedError as e:
       logging.exception('Unable to charge device to specified level.')
 
+  if options.max_battery_temp is not None:
+    try:
+      battery = battery_utils.BatteryUtils(device)
+      battery.LetBatteryCoolToTemperature(options.max_battery_temp)
+    except device_errors.CommandFailedError as e:
+      logging.exception('Unable to let battery cool to specified temperature.')
 
 def _ConfigureLocalProperties(device, java_debug=True):
   """Set standard readonly testing device properties prior to reboot."""
@@ -241,7 +247,7 @@ def _PushAndLaunchAdbReboot(device, target):
   # Launch adb_reboot
   logging.info('  Launching adb_reboot ...')
   device.RunShellCommand(
-      [device.GetDevicePieWrapper(), '/data/local/tmp/adb_reboot'],
+      ['/data/local/tmp/adb_reboot'],
       check_return=True)
 
 
@@ -308,6 +314,8 @@ def main():
                       help='list of adb keys to push to device')
   parser.add_argument('-v', '--verbose', action='count', default=1,
                       help='Log more information.')
+  parser.add_argument('--max-battery-temp', type=int, metavar='NUM',
+                      help='Wait for the battery to have this temp or lower.')
   args = parser.parse_args()
   constants.SetBuildType(args.target)