Theano (aesara) doesn't have theano.gof anymore

IvanYashchuk · Feb 28, 2021 · 3f165ef · 3f165ef
1 parent efa73ea
commit 3f165ef
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/fenics_pymc3/core.py b/fenics_pymc3/core.py
@@ -1,13 +1,19 @@
 import theano
-from theano.gof import Op, Apply
+
+try:
+    from theano.graph.op import Op, Apply, type
+except ModuleNotFoundError:
+    # older version of theano
+    from theano.gof import Op, Apply, type
+
 
 import functools
 
 from fecr import evaluate_primal, evaluate_pullback
 
 
 class FenicsVJPOp(Op):
-    params_type = theano.gof.type.Generic()
+    params_type = type.Generic()
     __props__ = ("ofunc", "templates", "fenics_output", "fenics_inputs", "tape")
 
     def __init__(self, ofunc, templates, fenics_output, fenics_inputs, tape):