Fix pickling errors with python 3.14

Author: Armavica

tests/link/c/test_cmodule.py

@@ -395,18 +395,18 @@ def test_linking_patch(listdir_mock, platform):
             ]
 
 
-def test_cache_race_condition():
-    with tempfile.TemporaryDirectory() as dir_name:
+@config.change_flags(on_opt_error="raise", on_shape_error="raise")
+def _f_build_cache_race_condition(factor):
+    # Some of the caching issues arise during constant folding within the
+    # optimization passes, so we need these config changes to prevent the
+    # exceptions from being caught
+    a = pt.vector()
+    f = pytensor.function([a], factor * a)
+    return f(np.array([1], dtype=config.floatX))
 
-        @config.change_flags(on_opt_error="raise", on_shape_error="raise")
-        def f_build(factor):
-            # Some of the caching issues arise during constant folding within the
-            # optimization passes, so we need these config changes to prevent the
-            # exceptions from being caught
-            a = pt.vector()
-            f = pytensor.function([a], factor * a)
-            return f(np.array([1], dtype=config.floatX))
 
+def test_cache_race_condition():
+    with tempfile.TemporaryDirectory() as dir_name:
         ctx = multiprocessing.get_context()
         compiledir_prop = pytensor.config._config_var_dict["compiledir"]
 
@@ -425,7 +425,7 @@ def f_build(factor):
                 # A random, constant input to prevent caching between runs
                 factor = rng.random()
                 procs = [
-                    ctx.Process(target=f_build, args=(factor,))
+                    ctx.Process(target=_f_build_cache_race_condition, args=(factor,))
                     for i in range(num_procs)
                 ]
                 for proc in procs:

tests/test_config.py

@@ -247,9 +247,12 @@ def test_config_pickling():
         configparser.IntParam(5, lambda i: i > 0),
         in_c_key=False,
     )
+    # Python 3.14 emits a pickle.PicklingError
+    # previous versions used to emit an AttributeError
+    # the error string changed a little bit too
     with pytest.raises(
-        AttributeError,
-        match=r"Can't (pickle|get) local object 'test_config_pickling\.<locals>\.<lambda>'",
+        (AttributeError, pickle.PicklingError),
+        match=r"Can't (pickle|get) local object .*test_config_pickling\.<locals>\.<lambda>",
     ):
         pickle.dump(root, io.BytesIO())