fix make_initial_point_fn

Author: ferrine

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "nutpie"
-version = "0.8.1"
+version = "0.8.2"
 authors = [
     "Adrian Seyboldt <[email protected]>",
     "PyMC Developers <[email protected]>"

pyproject.toml

@@ -30,10 +30,10 @@ dependencies = [
 
 [project.optional-dependencies]
 stan = ["bridgestan >= 2.1.2"]
-pymc = ["pymc >= 5.5.0", "numba >= 0.57.1"]
+pymc = ["pymc >= 5.6.0", "numba >= 0.57.1"]
 all = [
     "bridgestan >= 2.1.2",
-    "pymc >= 5.5.0",
+    "pymc >= 5.6.0",
     "numba >= 0.57.1",
 ]
 

python/nutpie/compile_pymc.py

@@ -15,6 +15,7 @@
 from numba import literal_unroll
 from numba.cpython.unsafe.tuple import alloca_once, tuple_setitem
 from numpy.typing import NDArray
+from pymc.initial_point import make_initial_point_fn
 
 from nutpie import _lib
 from nutpie.sample import CompiledModel
@@ -219,7 +220,7 @@ def compile_pymc_model(model: pm.Model, **kwargs) -> CompiledPyMCModel:
 
 
 def _compute_shapes(model):
-    point = pm.model.make_initial_point_fn(model=model, return_transformed=True)(0)
+    point = make_initial_point_fn(model=model, return_transformed=True)(0)
 
     trace_vars = {
         var.name: var