ChatGPT suggestions

Author: rwgk

cuda_bindings/tests/run_python_code_safely.py renamed to cuda_bindings/tests/spawned_process_runner.py

@@ -7,6 +7,7 @@
 import traceback
 from dataclasses import dataclass
 from io import StringIO
+from typing import Any, Callable, Optional, Sequence
 
 PROCESS_KILLED = -9
 PROCESS_NO_RESULT = -999
@@ -56,7 +57,14 @@ def __call__(self):
                 pass
 
 
-def run_in_spawned_child_process(target, *, args=None, kwargs=None, timeout=None, rethrow=False):
+def run_in_spawned_child_process(
+    target: Callable[..., None],
+    *,
+    args: Optional[Sequence[Any]] = None,
+    kwargs: Optional[dict[str, Any]] = None,
+    timeout: Optional[float] = None,
+    rethrow: bool = False,
+) -> CompletedProcess:
     """Run `target` in a spawned child process, capturing stdout/stderr.
 
     The provided `target` must be defined at the top level of a module, and must
@@ -100,7 +108,8 @@ def run_in_spawned_child_process(target, *, args=None, kwargs=None, timeout=None
         if rethrow and result.returncode != 0:
             raise ChildProcessError(
                 f"Child process exited with code {result.returncode}.\n"
-                f"--- stderr-from-child-process ---\n{result.stderr}"
+                "--- stderr-from-child-process ---\n"
+                f"{result.stderr}"
                 "<end-of-stderr-from-child-process>\n"
             )
 

cuda_bindings/tests/test_path_finder_load.py

@@ -5,7 +5,7 @@
 import sys
 
 import pytest
-import run_python_code_safely
+import spawned_process_runner
 
 from cuda.bindings import path_finder
 from cuda.bindings._path_finder import supported_libs
@@ -75,7 +75,7 @@ def test_find_or_load_nvidia_dynamic_library(info_summary_append, libname):
     # to ensure isolation of global dynamic linking state (e.g., dlopen handles).
     # Without child processes, loading/unloading libraries during testing could
     # interfere across test cases and lead to nondeterministic or platform-specific failures.
-    result = run_python_code_safely.run_in_spawned_child_process(child_process_func, args=(libname,), timeout=30)
+    result = spawned_process_runner.run_in_spawned_child_process(child_process_func, args=(libname,), timeout=30)
     if result.returncode == 0:
         info_summary_append(f"abs_path={result.stdout.rstrip()}")
     else: