Update tracing_new_process.py

apscheduler tries to schedule jobs when the interpreter is shutting down which can cause it to crash and leave us in a bad state

Author: KRRT7

codeflash/tracing/tracing_new_process.py

@@ -13,6 +13,7 @@
 import threading
 import time
 from collections import defaultdict
+from importlib.util import find_spec
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, Callable, ClassVar
 
@@ -47,6 +48,18 @@ def __init__(self, code: FakeCode, prior: FakeFrame | None) -> None:
         self.f_locals: dict = {}
 
 
+def patch_ap_scheduler() -> None:
+    if find_spec("apscheduler") is None:
+
+        import apscheduler.schedulers.background as bg
+        import apscheduler.schedulers.blocking as bb
+        from apscheduler.schedulers import base
+
+        bg.BackgroundScheduler.start = lambda _, *_a, **_k: None
+        bb.BlockingScheduler.start = lambda _, *_a, **_k: None
+        base.BaseScheduler.add_job = lambda _, *_a, **_k: None
+
+
 # Debug this file by simply adding print statements. This file is not meant to be debugged by the debugger.
 class Tracer:
     """Use this class as a 'with' context manager to trace a function call.
@@ -839,6 +852,7 @@ def runctx(self, cmd: str, global_vars: dict[str, Any], local_vars: dict[str, An
         }
     args_dict["config"]["module_root"] = Path(args_dict["config"]["module_root"])
     args_dict["config"]["tests_root"] = Path(args_dict["config"]["tests_root"])
+    patch_ap_scheduler()
     tracer = Tracer(
         config=args_dict["config"],
         output=Path(args_dict["output"]),