From 113b1996a920fb4735a8860d7f01c5a14a0ea037 Mon Sep 17 00:00:00 2001
From: Eric Johnson <ericdjohnson@google.com>
Date: Tue, 17 Sep 2024 10:56:49 -0700
Subject: [PATCH] No public description

PiperOrigin-RevId: 675630891
---
 google/colab/_shell.py | 5 +++++
 setup.py               | 2 +-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/google/colab/_shell.py b/google/colab/_shell.py
index 02a677b4..f602e2a6 100644
--- a/google/colab/_shell.py
+++ b/google/colab/_shell.py
@@ -26,6 +26,7 @@
 from ipykernel import jsonutil
 from ipykernel import zmqshell
 from IPython.core import alias
+from IPython.core import compilerop
 from IPython.core import inputsplitter
 from IPython.core import interactiveshell
 from IPython.core import oinspect
@@ -70,6 +71,10 @@ def init_history(self):
     self.history_manager = _history.ColabHistoryManager(shell=self, parent=self)
     self.configurables.append(self.history_manager)
 
+  def init_instance_attrs(self):
+    self.compiler_class = compilerop.CachingCompiler
+    super().init_instance_attrs()
+
   def _should_use_native_system_methods(self):
     # TODO: Update to match intended values, as appropriate.
     return bool(os.getenv('USE_NATIVE_IPYTHON_SYSTEM_COMMANDS'))
diff --git a/setup.py b/setup.py
index 1a477602..002354ce 100644
--- a/setup.py
+++ b/setup.py
@@ -20,7 +20,7 @@
     # Note: these dependency versions should be kept in-sync with the versions
     # specified in the docker container requirements files.
     'google-auth==2.27.0',
-    'ipykernel==5.5.6',
+    'ipykernel==6.17.1',
     'ipyparallel==8.8.0',
     'ipython==7.34.0',
     'notebook==6.5.5',