Remove unused argument from CpuLeapfrogIntegrator (#2445)

pymc-devs · Jul 28, 2017 · 1b6e9cc · 1b6e9cc
1 parent b988ba9
commit 1b6e9cc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/pymc3/step_methods/hmc/base_hmc.py b/pymc3/step_methods/hmc/base_hmc.py
@@ -65,5 +65,4 @@ def __init__(self, vars=None, scaling=None, step_scale=0.25, is_cov=False,
         else:
             self.potential = quad_potential(scaling, is_cov)
 
-        self.integrator = integration.CpuLeapfrogIntegrator(
-            size, self.potential, self._logp_dlogp_func)
+        self.integrator = integration.CpuLeapfrogIntegrator(self.potential, self._logp_dlogp_func)
diff --git a/pymc3/step_methods/hmc/integration.py b/pymc3/step_methods/hmc/integration.py
@@ -8,11 +8,8 @@
 
 
 class CpuLeapfrogIntegrator(object):
-    """Optimized leapfrog integration using numpy."""
-
-    def __init__(self, ndim, potential, logp_dlogp_func):
+    def __init__(self, potential, logp_dlogp_func):
         """Leapfrog integrator using CPU."""
-        self._ndim = ndim
         self._potential = potential
         self._logp_dlogp_func = logp_dlogp_func
         self._dtype = self._logp_dlogp_func.dtype