move it to its own module to avoid hound

better · Jul 7, 2019 · 4a862b3 · 4a862b3
1 parent 9c44cca
commit 4a862b3
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/convoys/autograd_scipy_monkeypatch.py b/convoys/autograd_scipy_monkeypatch.py
@@ -0,0 +1,8 @@
+# Monkey patching scipy to avoid issue with autograd
+# Newest scipy moved logsumexp from scipy.misc to scipy.special
+# This is fixed in latest autograd, but it's not on PyPI yet
+
+import scipy.misc
+if not hasattr(scipy.misc, 'logsumexp'):
+    import scipy.special
+    scipy.misc.logsumexp = scipy.special.logsumexp
diff --git a/convoys/regression.py b/convoys/regression.py
@@ -1,10 +1,4 @@
-# Monkey patching scipy to avoid issue with autograd
-import scipy.misc
-if not hasattr(scipy.misc, 'logsumexp'):
-    import scipy.special
-    scipy.misc.logsumexp = scipy.special.logsumexp
-
-# Now, continue with the real stuff.
+from convoys import autograd_scipy_monkeypatch
 import autograd
 from autograd_gamma import gammainc
 import emcee