From cfbe77b3675d940c2491e5f43302be40c5b21806 Mon Sep 17 00:00:00 2001
From: Xingyao Wang <xingyao@all-hands.dev>
Date: Fri, 20 Dec 2024 11:29:45 -0500
Subject: [PATCH] fix: only register atexit when EventStreamRuntime is
 initialized (#5712)

---
 openhands/runtime/impl/eventstream/eventstream_runtime.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/openhands/runtime/impl/eventstream/eventstream_runtime.py b/openhands/runtime/impl/eventstream/eventstream_runtime.py
index 384cfd5e4880..ae5bea2990c3 100644
--- a/openhands/runtime/impl/eventstream/eventstream_runtime.py
+++ b/openhands/runtime/impl/eventstream/eventstream_runtime.py
@@ -59,7 +59,7 @@ def remove_all_runtime_containers():
     remove_all_containers(CONTAINER_NAME_PREFIX)
 
 
-atexit.register(remove_all_runtime_containers)
+_atexit_registered = False
 
 
 class EventStreamRuntime(Runtime):
@@ -109,6 +109,11 @@ def __init__(
         attach_to_existing: bool = False,
         headless_mode: bool = True,
     ):
+        global _atexit_registered
+        if not _atexit_registered:
+            _atexit_registered = True
+            atexit.register(remove_all_runtime_containers)
+
         self.config = config
         self._host_port = 30000  # initial dummy value
         self._container_port = 30001  # initial dummy value