run-pre-commit

Author: Darshan808

ipykernel/zmqshell.py

@@ -26,11 +26,11 @@
 from IPython.core.autocall import ZMQExitAutocall
 from IPython.core.displaypub import DisplayPublisher
 from IPython.core.error import UsageError
+from IPython.core.history import HistoryOutput
 from IPython.core.interactiveshell import InteractiveShell, InteractiveShellABC
 from IPython.core.magic import Magics, line_magic, magics_class
 from IPython.core.magics import CodeMagics, MacroToEdit  # type:ignore[attr-defined]
 from IPython.core.usage import default_banner
-from IPython.core.history import HistoryOutput
 from IPython.display import Javascript, display
 from IPython.utils import openpy
 from IPython.utils.process import arg_split, system  # type:ignore[attr-defined]
@@ -116,7 +116,7 @@ def publish(  # type:ignore[override]
         update : bool, optional, keyword-only
             If True, send an update_display_data message instead of display_data.
         """
-        if self.shell is not None and hasattr(self.shell, 'history_manager'):
+        if self.shell is not None and hasattr(self.shell, "history_manager"):
             outputs = self.shell.history_manager.outputs
             outputs[self.shell.execution_count].append(
                 HistoryOutput(output_type="display_data", bundle=data)

tests/test_zmq_shell.py

@@ -221,7 +221,7 @@ def test_display_stored_in_history(self):
 
         self.disp_pub.shell = mock_shell
 
-        data = {'text/plain': 'test output'}
+        data = {"text/plain": "test output"}
         self.disp_pub.publish(data)
 
         # Check that output was stored in history
@@ -230,6 +230,7 @@ def test_display_stored_in_history(self):
         assert stored_outputs[0].output_type == "display_data"
         assert stored_outputs[0].bundle == data
 
+
 def test_magics(tmp_path):
     context = zmq.Context()
     socket = context.socket(zmq.PUB)