Fix docstring in battery_utils

build/android/pylib/device/battery_utils.py

 # Copyright 2015 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """Provides a variety of device interactions with power.
 """
 # pylint: disable=unused-argument
 
 import collections
 import contextlib
@@ skipping 182 matching lines @@
 
     Args:
       enabled: A boolean indicating whether charging should be enabled or
         disabled.
       timeout: timeout in seconds
       retries: number of retries
 
     Raises:
       device_errors.CommandFailedError: If method of disabling charging cannot
         be determined.
-      device_errors.DeviceVersionError: If device is not L or higher.
     """
     if 'charging_config' not in self._cache:
       for c in _CONTROL_CHARGING_COMMANDS:
         if self._device.FileExists(c['witness_file']):
           self._cache['charging_config'] = c
           break
       else:
         raise device_errors.CommandFailedError(
             'Unable to find charging commands.')
 
@@ skipping 50 matching lines @@
                                  check_return=True)
     timeout_retry.WaitFor(battery_updates_disabled, wait_period=1)
 
   # TODO(rnephew): Make private when all use cases can use the context manager.
   def EnableBatteryUpdates(self, timeout=None, retries=None):
     """ Restarts device charging so that dumpsys no longer collects power data.
 
     Args:
       timeout: timeout in seconds
       retries: number of retries
+
+    Raises:
+      device_errors.DeviceVersionError: If device is not L or higher.
     """
     def battery_updates_enabled():
       return self.GetCharging() is True
 
     if (self._device.build_version_sdk <
         constants.ANDROID_SDK_VERSION_CODES.LOLLIPOP):
       raise device_errors.DeviceVersionError('Device must be L or higher.')
 
     self._device.RunShellCommand(['dumpsys', 'battery', 'reset'],
                                  check_return=True)
@@ skipping 44 matching lines @@
       battery_level = self.GetBatteryInfo().get('level')
       if battery_level is None:
         logging.warning('Unable to find current battery level.')
         battery_level = 100
       else:
         logging.info('current battery level: %s', battery_level)
         battery_level = int(battery_level)
       return battery_level >= level
 
     timeout_retry.WaitFor(device_charged, wait_period=wait_period)