Move sampler code to nuts-rs and expose new features

Cargo.toml

@@ -21,22 +21,24 @@ name = "_lib"
 crate-type = ["cdylib"]
 
 [dependencies]
-nuts-rs = "0.8.0"
-numpy = "0.20.0"
+nuts-rs = "0.9.0"
+numpy = "0.21.0"
 ndarray = "0.15.6"
 rand = "0.8.5"
 thiserror = "1.0.44"
 rand_chacha = "0.3.1"
 rayon = "1.9.0"
-arrow2 = "0.17.0"
+arrow2 = "0.18.0"
 anyhow = "1.0.72"
 itertools = "0.12.0"
 bridgestan = "2.1.2"
 rand_distr = "0.4.3"
 smallvec = "1.11.0"
+upon = { version = "0.8.1", default-features = false, features = [] }
+time-humanize = { version = "0.1.3", default-features = false }
 
 [dependencies.pyo3]
-version = "0.20.0"
+version = "0.21.0"
 features = ["extension-module", "anyhow"]
 
 [dev-dependencies]

python/nutpie/compile_pymc.py

@@ -87,9 +87,11 @@ def with_data(self, **updates):
             user_data=user_data,
         )
 
-    def _make_sampler(self, settings, init_mean, chains, cores, seed, callback=None):
+    def _make_sampler(self, settings, init_mean, cores, template, rate, callback=None):
         model = self._make_model(init_mean)
-        return _lib.PySampler.from_pymc(settings, chains, cores, model, seed, callback)
+        return _lib.PySampler.from_pymc(
+            settings, cores, model, template, rate, callback
+        )
 
     def _make_model(self, init_mean):
         expand_fn = _lib.ExpandFunc(

python/nutpie/compile_stan.py

@@ -80,9 +80,11 @@ def _make_model(self, init_mean):
             return self.with_data().model
         return self.model
 
-    def _make_sampler(self, settings, init_mean, chains, cores, seed, callback=None):
+    def _make_sampler(self, settings, init_mean, cores, template, rate, callback=None):
         model = self._make_model(init_mean)
-        return _lib.PySampler.from_stan(settings, chains, cores, model, seed, callback)
+        return _lib.PySampler.from_stan(
+            settings, cores, model, template, rate, callback
+        )
 
     @property
     def n_dim(self):