Fix imports

IvanYashchuk · Feb 28, 2021 · fd6ff26 · fd6ff26
1 parent 31ca1cf
commit fd6ff26
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/fenics_pymc3/core.py b/fenics_pymc3/core.py
@@ -1,7 +1,8 @@
 import theano
 
 try:
-    from theano.graph.op import Op, Apply, type
+    from theano.graph.op import Op, Apply
+    from theano.graph import type
 except ModuleNotFoundError:
     # older version of theano
     from theano.gof import Op, Apply, type