Rename LocalOptTracker to OpToRewriterTracker

aesara/graph/opt.py

@@ -1152,8 +1152,8 @@ def node_rewriter(
     return decorator
 
 
-class LocalOptTracker:
-    r"""A container that maps rewrites to `Op` instances and `Op`-type inheritance."""
+class OpToRewriterTracker:
+    r"""A container that maps `NodeRewriter`\s to `Op` instances and `Op`-type inheritance."""
 
     def __init__(self):
         self.tracked_instances: Dict[Op, List[NodeRewriter]] = {}
@@ -1256,7 +1256,7 @@ class LocalOptGroup(NodeRewriter):
             self.applied_true: Dict[Rewriter, int] = {}
             self.node_created: Dict[Rewriter, int] = {}
 
-        self.tracker = LocalOptTracker()
+        self.tracker = OpToRewriterTracker()
 
         for o in self.opts:
 
@@ -2246,7 +2246,7 @@ class EquilibriumOptimizer(NavigatorOptimizer):
         self.global_optimizers: List[GraphRewriter] = []
         self.tracks_on_change_inputs = tracks_on_change_inputs
 
-        self.node_tracker = LocalOptTracker()
+        self.node_tracker = OpToRewriterTracker()
 
         for opt in optimizers:
             if isinstance(opt, NodeRewriter):
@@ -3163,6 +3163,11 @@ DEPRECATED_NAMES = [
         "`FromFunctionLocalOptimizer` is deprecated: use `FromFunctionNodeRewriter` instead.",
         FromFunctionNodeRewriter,
     ),
+    (
+        "LocalOptTracker",
+        "`LocalOptTracker` is deprecated: use `OpToRewriterTracker` instead.",
+        OpToRewriterTracker,
+    ),
 ]
 
 

tests/graph/test_opt.py

@@ -8,10 +8,10 @@ from aesara.graph.op import Op
 from aesara.graph.opt import (
     EquilibriumOptimizer,
     LocalOptGroup,
-    LocalOptTracker,
     MergeOptimizer,
     OpKeyOptimizer,
     OpSub,
+    OpToRewriterTracker,
     PatternSub,
     TopoOptimizer,
     in2out,
@@ -766,7 +766,7 @@ def test_node_rewriter():
     assert hits[0] == 2
 
 
-def test_TrackingNodeRewriter():
+def test_OpToRewriterTracker():
     @node_rewriter(None)
     def local_opt_1(fgraph, node):
         pass
@@ -787,7 +787,7 @@ def test_TrackingNodeRewriter():
     def local_opt_5(fgraph, node):
         pass
 
-    tracker = LocalOptTracker()
+    tracker = OpToRewriterTracker()
     tracker.add_tracker(local_opt_1)
     tracker.add_tracker(local_opt_2)
     tracker.add_tracker(local_opt_3)