replace the closure in wrap_with_catch with a callable class

Author: SimonBlanke

src/gradient_free_optimizers/_catch.py

@@ -6,18 +6,23 @@
 logger = logging.getLogger(__name__)
 
 
-def wrap_with_catch(
-    objective_function: Callable,
-    catch: dict[type[Exception], int | float],
-) -> Callable:
-    """Wrap objective function to catch exceptions and return fallback scores."""
-    catch_types = tuple(catch.keys())
+class _CatchWrapper:
+    """Callable that catches exceptions from the objective and returns fallbacks.
+
+    A class instead of a closure so it can be pickled by multiprocessing's
+    spawn context (Windows, macOS 3.14+).
+    """
+
+    def __init__(self, func: Callable, catch: dict[type[Exception], int | float]):
+        self.func = func
+        self.catch = catch
+        self._catch_types = tuple(catch.keys())
 
-    def wrapped(params):
+    def __call__(self, params):
         try:
-            return objective_function(params)
-        except catch_types as e:
-            for exc_type, fallback_score in catch.items():
+            return self.func(params)
+        except self._catch_types as e:
+            for exc_type, fallback_score in self.catch.items():
                 if isinstance(e, exc_type):
                     logger.warning(
                         "Caught %s in objective function: %s. "
@@ -29,4 +34,10 @@ def wrapped(params):
                     return fallback_score
             raise
 
-    return wrapped
+
+def wrap_with_catch(
+    objective_function: Callable,
+    catch: dict[type[Exception], int | float],
+) -> Callable:
+    """Wrap objective function to catch exceptions and return fallback scores."""
+    return _CatchWrapper(objective_function, catch)