Internal change

PiperOrigin-RevId: 514821183

Author: metrizable

google/__init__.py

@@ -16,7 +16,9 @@
 # pylint:disable=g-import-not-at-top
 try:
   import pkg_resources
+
   pkg_resources.declare_namespace(__name__)
 except ImportError:
   import pkgutil
+
   __path__ = pkgutil.extend_path(__path__, __name__)

google/colab/__init__.py

@@ -31,8 +31,14 @@
 from google.colab import widgets
 
 __all__ = [
-    'auth', 'data_table', 'drive', 'files', 'output', 'runtime', 'snippets',
-    'widgets'
+    'auth',
+    'data_table',
+    'drive',
+    'files',
+    'output',
+    'runtime',
+    'snippets',
+    'widgets',
 ]
 
 __version__ = '0.0.1a2'

google/colab/_debugpy.py

@@ -66,8 +66,8 @@ def inspector_thread():
       # Need to start this thread after the debugpy.listen() call but before
       # the undo_patch_thread_modules().
       threading.Thread(
-          target=inspector_thread, name='_colab_inspector_thread',
-          daemon=True).start()
+          target=inspector_thread, name='_colab_inspector_thread', daemon=True
+      ).start()
       inspector_thread_started.wait()
 
     # Debugger tracing isn't needed for just tracebacks, but if full debugging
@@ -81,10 +81,12 @@ def inspector_thread():
     try:
       # Stop debugpy from tracing newly created threads.
       from _pydev_bundle import pydev_monkey  # pylint: disable=g-import-not-at-top
+
       pydev_monkey.undo_patch_thread_modules()
     except ModuleNotFoundError:
       # _pydev_bundle may be vendored into either location.
       from pydevd._pydev_bundle import pydev_monkey  # pylint: disable=g-import-not-at-top
+
       pydev_monkey.undo_patch_thread_modules()
 
     # Clear the trace flag to allow fetching stack traces.

google/colab/_debugpy_repr.py

@@ -52,9 +52,12 @@ def get_shape(obj):
   if isinstance(obj, collections_abc.Sized):
     try:
       shape = getattr(obj, 'shape', None)
-      if (isinstance(shape, tuple) or
-          hasattr(shape, '__module__') and isinstance(shape.__module__, str) and
-          'tensorflow.' in shape.__module__):
+      if (
+          isinstance(shape, tuple)
+          or hasattr(shape, '__module__')
+          and isinstance(shape.__module__, str)
+          and 'tensorflow.' in shape.__module__
+      ):
         return str(shape)
     except Exception:  # pylint: disable=broad-except
       logging.exception('Unexpected exception finding object shape')

google/colab/_history.py

@@ -26,6 +26,7 @@ class ColabHistoryManager(history.HistoryManager):
   This allows us to associate code executions with the cell which was executed
   in Colab's UI.
   """
+
   _input_hist_cells = [{'code': '', 'cell_id': '', 'start_time': 0}]
 
   def reset(self, new_session=True):
@@ -36,12 +37,16 @@ def reset(self, new_session=True):
   def store_inputs(self, line_num, source, source_raw=None):
     """Variant of HistoryManager.store_inputs which also stores the cell ID."""
     super(ColabHistoryManager, self).store_inputs(
-        line_num, source, source_raw=source_raw)
+        line_num, source, source_raw=source_raw
+    )
 
     # The parent_header on the shell is the message that resulted in the code
     # execution request. Grab the cell ID out of that.
-    cell_id = self.shell.parent_header.get('metadata',
-                                           {}).get('colab', {}).get('cell_id')
+    cell_id = (
+        self.shell.parent_header.get('metadata', {})
+        .get('colab', {})
+        .get('cell_id')
+    )
 
     self._input_hist_cells.append({
         'code': source_raw,
@@ -76,10 +81,10 @@ def _executed_cells_as_json(self, include_source_hash=False):
       if include_source_hash:
         # LINT.IfChange(execution_count)
         cells[cell['cell_id']] = {
-            'executionCount':
-                i,
-            'sourceHash':
-                hashlib.md5(cell['code'].encode('utf8')).hexdigest()[:10]
+            'executionCount': i,
+            'sourceHash': hashlib.md5(cell['code'].encode('utf8')).hexdigest()[
+                :10
+            ],
         }
         # LINT.ThenChange()
       else:

google/colab/_import_hooks/_altair.py

@@ -24,7 +24,9 @@ class _AltairImportHook:
 
   def find_module(self, fullname, path=None):
     if fullname not in [
-        'altair.vegalite.v2', 'altair.vegalite.v3', 'altair.vegalite.v4'
+        'altair.vegalite.v2',
+        'altair.vegalite.v3',
+        'altair.vegalite.v4',
     ]:
       return None
     self.module_info = imp.find_module(fullname.split('.')[-1], path)

google/colab/_import_hooks/_bokeh.py

@@ -51,7 +51,8 @@ def load_module(self, name):
         #     output_notebook()
         # just works without modification.
         _bokeh_io_module.notebook.install_notebook_hook(
-            'jupyter', _load_notebook, _show_doc, _show_app, overwrite=True)
+            'jupyter', _load_notebook, _show_doc, _show_app, overwrite=True
+        )
       except:  # pylint: disable=bare-except
         logging.exception('Error enabling Bokeh Colab rendering.')
         os.environ['COLAB_BOKEH_IMPORT_HOOK_EXCEPTION'] = '1'
@@ -77,17 +78,17 @@ def _show_doc(obj, state, notebook_handle):
     _bokeh_loaded_in_this_cell = True
     IPython.get_ipython().events.register('post_run_cell', _post_execute)  # pylint: disable=undefined-variable
     _bokeh_io_module.notebook.load_notebook(
-        resources=_bokeh_resources, hide_banner=True)
+        resources=_bokeh_resources, hide_banner=True
+    )
 
   # Call the default bokeh rendering path.
   return _bokeh_io_module.notebook.show_doc(obj, state, notebook_handle)
 
 
 # pylint: disable=unused-argument
-def _load_notebook(resources=None,
-                   verbose=False,
-                   hide_banner=False,
-                   load_timeout=5000):
+def _load_notebook(
+    resources=None, verbose=False, hide_banner=False, load_timeout=5000
+):
   global _bokeh_resources
   # In Jupyter this method is called once per notebook launch but with Colab's
   # isolated outputframes this loading needs to be done for each outputframe.

google/colab/_import_hooks/_client_info.py

@@ -38,6 +38,7 @@ def load_module(self, fullname):
 
     Args:
       fullname: fullname of the module
+
     Returns:
       A modified google.api_core.client_info module.
     """

google/colab/_import_hooks/_cv2.py

@@ -60,7 +60,8 @@ class _OpenCVImportHook:
       '{0}.imshow() is disabled in Colab, because it causes Jupyter sessions\n'
       'to crash; see https://github.com/jupyter/notebook/issues/3935.\n'
       'As a substitution, consider using\n'
-      '  from google.colab.patches import {0}_imshow\n')
+      '  from google.colab.patches import {0}_imshow\n'
+  )
   env_var = 'ENABLE_CV2_IMSHOW'
 
   def find_module(self, fullname, path=None):
@@ -82,7 +83,8 @@ def load_module(self, name):
             cv_module.imshow,
             message=self.message.format(name),
             env_var=self.env_var,
-            name='{}.imshow'.format(name))
+            name='{}.imshow'.format(name),
+        )
       except:  # pylint: disable=bare-except
         logging.exception('Error disabling %s.imshow().', name)
         os.environ['COLAB_CV2_IMPORT_HOOK_EXCEPTION'] = '1'

google/colab/_import_hooks/_pydrive.py

@@ -43,20 +43,26 @@ def load_module(self, name):
       try:
         import httplib2  # pylint:disable=g-import-not-at-top
         from oauth2client.contrib.gce import AppAssertionCredentials  # pylint:disable=g-import-not-at-top
-        orig_local_webserver_auth = pydrive_auth_module.GoogleAuth.LocalWebserverAuth
+
+        orig_local_webserver_auth = (
+            pydrive_auth_module.GoogleAuth.LocalWebserverAuth
+        )
 
         # Capture the environment variable outside of the patched method since
         # self will refer to a GoogleAuth object in these cases.
         env_var = self.env_var
 
         def PatchedLocalWebServerAuth(self, *args, **kwargs):  # pylint:disable=invalid-name
           if not os.environ.get(env_var, '') and isinstance(
-              self.credentials, AppAssertionCredentials):
+              self.credentials, AppAssertionCredentials
+          ):
             self.credentials.refresh(httplib2.Http())
             return
           return orig_local_webserver_auth(self, *args, **kwargs)
 
-        pydrive_auth_module.GoogleAuth.LocalWebserverAuth = PatchedLocalWebServerAuth
+        pydrive_auth_module.GoogleAuth.LocalWebserverAuth = (
+            PatchedLocalWebServerAuth
+        )
       except:  # pylint: disable=bare-except
         logging.exception('Error patching PyDrive')