From b76fa94abae4505d0d5b14dcf1d77521c02482e4 Mon Sep 17 00:00:00 2001
From: Guilherme Ruiz <57478888+guigoruiz1@users.noreply.github.com>
Date: Sun, 16 Jun 2024 08:55:39 +0100
Subject: [PATCH] {#179|BUG FIX}: fix for ipython>8.17.0 (#180)

* fix for ipython>8.17.0

* Update halo_notebook.py

Close widget on clear

---------

Co-authored-by: Manraj Singh <manrajsinghgrover@gmail.com>
---
 halo/halo.py          | 2 +-
 halo/halo_notebook.py | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/halo/halo.py b/halo/halo.py
index cab2185..e25b250 100644
--- a/halo/halo.py
+++ b/halo/halo.py
@@ -99,7 +99,7 @@ def __init__(
 
         environment = get_environment()
 
-        def clean_up():
+        def clean_up(args):
             """Handle cell execution"""
             self.stop()
 
diff --git a/halo/halo_notebook.py b/halo/halo_notebook.py
index d2c8c91..3bf097b 100644
--- a/halo/halo_notebook.py
+++ b/halo/halo_notebook.py
@@ -53,6 +53,7 @@ def clear(self):
             self.output.outputs += self._output(self.CLEAR_LINE)
 
         self.output.outputs = self._output()
+        self.output.close()
         return self
 
     def _render_frame(self):