Remove numdifftools

Author: ColCarroll

pymc3/tests/test_distributions.py

@@ -471,37 +471,9 @@ def check_int_to_1(self, model, value, domain, paramdomains):
             area = integrate_nd(pdfx, domain, value.dshape, value.dtype)
             assert_almost_equal(area, 1, err_msg=str(pt))
 
-    def check_dlogp(self, model, value, domain, paramdomains):
-        try:
-            from numdifftools import Gradient
-        except ImportError:
-            return
-        if not model.cont_vars:
-            return
-
-        domains = paramdomains.copy()
-        domains['value'] = domain
-        bij = DictToArrayBijection(
-            ArrayOrdering(model.cont_vars), model.test_point)
-        dlogp = bij.mapf(model.fastdlogp(model.cont_vars))
-        logp = bij.mapf(model.fastlogp)
-
-        def wrapped_logp(x):
-            try:
-                return logp(x)
-            except:
-                return np.nan
-
-        ndlogp = Gradient(wrapped_logp)
-        for pt in product(domains, n_samples=100):
-            pt = Point(pt, model=model)
-            pt = bij.map(pt)
-            decimals = select_by_precision(float64=6, float32=4)
-            assert_almost_equal(dlogp(pt), ndlogp(pt), decimal=decimals, err_msg=str(pt))
-
     def checkd(self, distfam, valuedomain, vardomains, checks=None, extra_args=None):
         if checks is None:
-            checks = (self.check_int_to_1, self.check_dlogp)
+            checks = (self.check_int_to_1, )
 
         if extra_args is None:
             extra_args = {}
@@ -940,7 +912,8 @@ def test_wishart(self, n):
         # This check compares the autodiff gradient to the numdiff gradient.
         # However, due to the strict constraints of the wishart,
         # it is impossible to numerically determine the gradient as a small
-        # pertubation breaks the symmetry. Thus disabling.
+        # pertubation breaks the symmetry. Thus disabling. Also, numdifftools was
+        # removed in June 2019, so an alternative would be needed.
         #
         # self.checkd(Wishart, PdMatrix(n), {'n': Domain([2, 3, 4, 2000]), 'V': PdMatrix(n)},
         #             checks=[self.check_dlogp])
@@ -1120,12 +1093,6 @@ def logp(x):
             return -log(2 * .5) - abs(x - .5) / .5
         self.checkd(DensityDist, R, {}, extra_args={'logp': logp})
 
-    def test_addpotential(self):
-        with Model() as model:
-            value = Normal('value', 1, 1)
-            Potential('value_squared', -value ** 2)
-            self.check_dlogp(model, value, R, {})
-
     def test_get_tau_sigma(self):
         sigma = np.array([2])
         assert_almost_equal(continuous.get_tau_sigma(sigma=sigma), [1. / sigma**2, sigma])

pymc3/tuning/__init__.py

@@ -1,2 +1,2 @@
 from .starting import find_MAP
-from .scaling import approx_hessian, find_hessian, trace_cov, guess_scaling
+from .scaling import find_hessian, trace_cov, guess_scaling

pymc3/tuning/scaling.py

@@ -4,41 +4,7 @@
 from ..theanof import hessian_diag, inputvars
 from ..blocking import DictToArrayBijection, ArrayOrdering
 
-__all__ = ['approx_hessian', 'find_hessian', 'trace_cov', 'guess_scaling']
-
-
-def approx_hessian(point, vars=None, model=None):
-    """
-    Returns an approximation of the Hessian at the current chain location.
-
-    Parameters
-    ----------
-    model : Model (optional if in `with` context)
-    point : dict
-    vars : list
-        Variables for which Hessian is to be calculated.
-    """
-    from numdifftools import Jacobian
-
-    model = modelcontext(model)
-    if vars is None:
-        vars = model.cont_vars
-    vars = inputvars(vars)
-
-    point = Point(point, model=model)
-
-    bij = DictToArrayBijection(ArrayOrdering(vars), point)
-    dlogp = bij.mapf(model.fastdlogp(vars))
-
-    def grad_logp(point):
-        return np.nan_to_num(dlogp(point))
-
-    '''
-    Find the jacobian of the gradient function at the current position
-    this should be the Hessian; invert it to find the approximate
-    covariance matrix.
-    '''
-    return -Jacobian(grad_logp)(bij.map(point))
+__all__ = ['find_hessian', 'trace_cov', 'guess_scaling']
 
 
 def fixed_hessian(point, vars=None, model=None):

requirements-dev.txt

@@ -7,7 +7,6 @@ Keras>=2.0.8
 nbsphinx>=0.2.13
 nose>=1.3.7
 nose-parameterized==0.6.0
-numdifftools>=0.9.20
 numpy>=1.13.0
 numpydoc==0.7.0
 pycodestyle>=2.3.1