feat: Report detailed progress

Author: aseyboldt

Cargo.lock

@@ -34,6 +34,8 @@ 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.21.0"

Cargo.toml

@@ -87,9 +87,11 @@ def with_data(self, **updates):
             user_data=user_data,
         )
 
-    def _make_sampler(self, settings, init_mean, cores, 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, cores, model, 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_pymc.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, cores, 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, cores, model, callback)
+        return _lib.PySampler.from_stan(
+            settings, cores, model, template, rate, callback
+        )
 
     @property
     def n_dim(self):