Change use of "opt[imize]" to "rewrite"

Author: brandonwillard

aehmc/utils.py

@@ -4,8 +4,8 @@
 import aesara.tensor as at
 from aesara.graph.basic import Variable, ancestors
 from aesara.graph.fg import FunctionGraph
-from aesara.graph.opt_utils import optimize_graph
-from aesara.tensor.basic_opt import ShapeFeature
+from aesara.graph.rewriting.utils import rewrite_graph
+from aesara.tensor.rewriting.shape import ShapeFeature
 from aesara.tensor.var import TensorVariable
 
 
@@ -16,7 +16,7 @@ def simplify_shapes(graphs: List[Variable]):
         features=[ShapeFeature()],
         clone=False,
     )
-    return optimize_graph(shape_fg).outputs
+    return rewrite_graph(shape_fg).outputs
 
 
 class RaveledParamsMap:

environment.yml

@@ -11,8 +11,8 @@ dependencies:
   - compilers
   - numpy>=1.18.1
   - scipy>=1.4.0
-  - aesara>=2.7.5
-  - aeppl>=0.0.9
+  - aesara>=2.8.0
+  - aeppl>=0.0.35
   # Intel BLAS
   - mkl
   - mkl-service

setup.py

@@ -17,8 +17,8 @@
     install_requires=[
         "numpy>=1.18.1",
         "scipy>=1.4.0",
-        "aesara>=2.7.5",
-        "aeppl>=0.0.9",
+        "aesara>=2.8.0",
+        "aeppl>=0.0.35",
     ],
     tests_require=["pytest"],
     long_description=open("README.md").read() if exists("README.md") else "",