Fixing keyword args in top-level pm.compile_fn

`Model.compile_fn` requires all parameters except `outs` to be keyword arguments so the top level `compile_fn` didn't work as written.

Author: dfm

pymc/model.py

@@ -1886,7 +1886,7 @@ def compile_fn(outs, mode=None, point_fn=True, model=None, **kwargs):
     Compiled Aesara function as point function.
     """
     model = modelcontext(model)
-    return model.compile_fn(outs, mode, point_fn=point_fn, **kwargs)
+    return model.compile_fn(outs, mode=mode, point_fn=point_fn, **kwargs)
 
 
 def Point(*args, filter_model_vars=False, **kwargs) -> Dict[str, np.ndarray]:

pymc/tests/test_model.py

@@ -996,3 +996,21 @@ def test_deterministic():
 
 def test_empty_model_representation():
     assert pm.Model().str_repr() == ""
+
+
+def test_compile_fn():
+    with pm.Model() as m:
+        x = pm.Normal("x", 0, 1, size=2)
+        y = pm.LogNormal("y", 0, 1, size=2)
+
+    test_vals = np.array([0.0, -1.0])
+    state = {"x": test_vals, "y": test_vals}
+
+    with m:
+        func = pm.compile_fn(x + y, inputs=[x, y])
+        result_compute = func(state)
+
+    func = m.compile_fn(x + y, inputs=[x, y])
+    result_expect = func(state)
+
+    np.testing.assert_allclose(result_compute, result_expect)