rename execution method of the routine

rlouf · Apr 3, 2020 · db62a78 · db62a78
1 parent 109cedf
commit db62a78
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mcx/execution.py b/mcx/execution.py
@@ -22,7 +22,7 @@ def __init__(self, rng_key, runtime, num_warmup=1000, num_chains=4, **kwargs):
         self.unravel_fn = unravel_fn
         self.to_trace = to_trace
 
-    def take(self, num_samples=1000):
+    def run(self, num_samples=1000):
         _, self.rng_key = jax.random.split(self.rng_key)
 
         @jax.jit