unregister _kill_process after closing process

BrianPugh · Jan 30, 2023 · 08db9c6 · 08db9c6
1 parent fa37bf3
commit 08db9c6
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/belay/pyboard.py b/belay/pyboard.py
@@ -220,13 +220,11 @@ def process_output():
             # Windows needs more time
             time.sleep(5.0)
 
-        def cleanup():
-            _kill_process(subp.pid)
-
-        atexit.register(cleanup)
+        atexit.register(self.close)
 
     def close(self):
         _kill_process(self.subp.pid)
+        atexit.unregister(self.close)
 
     def read(self, size=1):
         while len(self.buf) < size: