pymc-devs
diff --git a/‎Cargo.lock
Lines changed: 96 additions & 42 deletions b/‎Cargo.lock
Lines changed: 96 additions & 42 deletions
diff --git a/‎Cargo.toml
Lines changed: 3 additions & 2 deletions b/‎Cargo.toml
Lines changed: 3 additions & 2 deletions
diff --git a/‎python/nutpie/compile_pymc.py
Lines changed: 5 additions & 2 deletions b/‎python/nutpie/compile_pymc.py
Lines changed: 5 additions & 2 deletions
@@ -21,22 +21,23 @@ name = "_lib"
 crate-type = ["cdylib"]
 
 [dependencies]
-nuts-rs = "0.10.0"
+nuts-rs = "0.11.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"
 # Keep arrow in sync with nuts-rs requirements
-arrow = { version = "51.0.0", default-features = false, features = ["ffi"] }
+arrow = { version = "52.0.0", default-features = false, features = ["ffi"] }
 anyhow = "1.0.72"
 itertools = "0.13.0"
 bridgestan = "2.4.1"
 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 }
+indicatif = "0.17.8"
 
 [dependencies.pyo3]
 version = "0.21.0"
 
@@ -87,10 +87,13 @@ def with_data(self, **updates):
             user_data=user_data,
         )
 
-    def _make_sampler(self, settings, init_mean, cores, template, rate, callback=None):
+    def _make_sampler(self, settings, init_mean, cores, progress_type):
         model = self._make_model(init_mean)
         return _lib.PySampler.from_pymc(
-            settings, cores, model, template, rate, callback
+            settings,
+            cores,
+            model,
+            progress_type,
         )
 
     def _make_model(self, init_mean):
Original file line number	Diff line number	Diff line change
`@@ -87,10 +87,13 @@ def with_data(self, **updates):`
`87`	`87`	`user_data=user_data,`
`88`	`88`	`)`
`89`	`89`
`90`		`- def _make_sampler(self, settings, init_mean, cores, template, rate, callback=None):`
	`90`	`+ def _make_sampler(self, settings, init_mean, cores, progress_type):`
`91`	`91`	`model = self._make_model(init_mean)`
`92`	`92`	`return _lib.PySampler.from_pymc(`
`93`		`- settings, cores, model, template, rate, callback`
	`93`	`+ settings,`
	`94`	`+ cores,`
	`95`	`+ model,`
	`96`	`+ progress_type,`
`94`	`97`	`)`
`95`	`98`
`96`	`99`	`def _make_model(self, init_mean):`