piiswrong closed pull request #10961: [MXNET-10896] [MXNET-10901] Fix test_sparse_mathematical_core sensitivity to scipy v1.1 URL: https://github.com/apache/incubator-mxnet/pull/10961
This is a PR merged from a forked repository. As GitHub hides the original diff on merge, it is displayed below for the sake of provenance: As this is a foreign pull request (from a fork), the diff is supplied below (as it won't show otherwise due to GitHub magic): diff --git a/tests/python/unittest/test_sparse_operator.py b/tests/python/unittest/test_sparse_operator.py index 1b3c128995a..226db70a2ac 100644 --- a/tests/python/unittest/test_sparse_operator.py +++ b/tests/python/unittest/test_sparse_operator.py @@ -1044,12 +1044,16 @@ def check_mathematical_core(stype, output_grad_stype=None, try: from scipy import special as scipy_special - import_succeeded = True + # On scipy v1.0, psi([0, -1, -2, -3, ...]) = [ inf, inf, inf, inf, ...] + # On scipy v1.1, psi([0, -1, -2, -3, ...]) = [-inf, nan, nan, nan, ...] + # Map the behavior of v1.1 psi() to that of v1.0 for ints <= 0 for consistency + scipy_psi = np.vectorize(lambda x: np.inf if float(x).is_integer() and x <= 0 else + scipy_special.psi(x)) # gamma check_sparse_mathematical_core("gamma", stype, lambda x: mx.sym.sparse.gamma(x), lambda x: scipy_special.gamma(x), - lambda x: scipy_special.gamma(x) * scipy_special.psi(x), + lambda x: scipy_special.gamma(x) * scipy_psi(x), output_grad_stype=output_grad_stype, input_grad_stype=input_grad_stype, force_overlap=force_overlap, @@ -1058,17 +1062,14 @@ def check_mathematical_core(stype, output_grad_stype=None, check_sparse_mathematical_core("gammaln", stype, lambda x: mx.sym.sparse.gammaln(x), lambda x: scipy_special.gammaln(x), - lambda x: scipy_special.psi(x), + lambda x: scipy_psi(x), output_grad_stype=output_grad_stype, input_grad_stype=input_grad_stype, force_overlap=force_overlap, density=density, ograd_density=ograd_density) - except: - if import_succeeded == False: - print("Could not import scipy. Skipping unit tests for special functions") - else: - raise + except ImportError: + print("Could not import scipy. Skipping unit tests for special functions") for i in range(1): print("pass", i) ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services