reverts

Author: 1stprinciple

development/utils/subprocess_manager.py

@@ -2,8 +2,7 @@
 import os
 import re  # Added for Serveo URL parsing
 import shutil  # Added for checking ssh availability
-
-# import signal
+import signal
 import subprocess
 import time
 from typing import IO, Any, Dict, List, Optional  # Added IO, Any, List, Dict, Optional

eval_protocol/cli_commands/deploy.py

@@ -19,15 +19,13 @@
     # Import functions with explicit names to match expected signatures
     from development.utils.subprocess_manager import (
         start_ngrok_and_get_url as _start_ngrok_and_get_url,
-    )
-    from development.utils.subprocess_manager import start_process as _start_process
-    from development.utils.subprocess_manager import (
+        start_process as _start_process,
         start_serveo_and_get_url as _start_serveo_and_get_url,
+        stop_process as _stop_process,
     )
-    from development.utils.subprocess_manager import stop_process as _stop_process
 except ImportError:
     # Fallback implementations when development module is not available
-    # import signal
+    import signal
     import socket
     import subprocess
 
@@ -81,9 +79,12 @@ def start_ngrok_and_get_url(local_port, log_path):
 
 
 from eval_protocol.auth import get_fireworks_account_id
-from eval_protocol.config import GCPCloudRunConfig, RewardKitConfig
-from eval_protocol.config import _config_file_path as global_loaded_config_path
-from eval_protocol.config import get_config
+from eval_protocol.config import (
+    GCPCloudRunConfig,
+    RewardKitConfig,
+    _config_file_path as global_loaded_config_path,
+    get_config,
+)
 from eval_protocol.evaluation import create_evaluation
 from eval_protocol.gcp_tools import (
     build_and_push_docker_image,

eval_protocol/pytest/default_mcp_gym_rollout_processor.py

@@ -1,8 +1,10 @@
 import asyncio
 import atexit
 import os
+import signal
 import socket
 import subprocess
+import threading
 import time
 from pathlib import Path
 from typing import List, Optional
@@ -180,8 +182,9 @@ def _signal_handler(cls, signum, frame):
     def _register_cleanup_handlers(cls):
         """Register cleanup handlers - called only once"""
         atexit.register(cls._cleanup_all_servers)
-        # signal.signal(signal.SIGINT, cls._signal_handler)  # Ctrl+C
-        # signal.signal(signal.SIGTERM, cls._signal_handler)  # Termination signal
+        if threading.current_thread() is threading.main_thread():
+            signal.signal(signal.SIGINT, cls._signal_handler)  # Ctrl+C
+            signal.signal(signal.SIGTERM, cls._signal_handler)  # Termination signal
 
     def __enter__(self):
         """Context manager entry"""

eval_protocol/rewards/apps_testing_util.py

@@ -23,7 +23,7 @@
 import re  # Added for re.search
 
 # to run the solution files we're using a timing based approach
-# import signal
+import signal
 import sys
 import textwrap  # Added for dedenting model output
 import traceback

eval_protocol/rewards/code_execution.py

@@ -21,8 +21,7 @@
 import re
 import resource
 import shlex  # Added for robust splitting of arguments
-
-# import signal
+import signal
 import subprocess
 import sys
 import tempfile