Rename support point rewrite

pymc-devs · Jun 24, 2024 · 5935250 · 5935250
1 parent 5dd4e46
commit 5935250
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pymc/distributions/custom.py b/pymc/distributions/custom.py
@@ -395,7 +395,7 @@ def _infer_final_signature(signature: str, n_inputs, n_outputs, n_rngs) -> str:
         return signature
 
 
-class FiniteLogpPointRewrite(GraphRewriter):
+class SupportPointRewrite(GraphRewriter):
     def rewrite_support_point_scan_node(self, node):
         if not isinstance(node.op, Scan):
             return
@@ -454,7 +454,7 @@ def dist_support_point(op, rv, *args):
     rv_out_idx = node.outputs.index(rv)
 
     fgraph = op.fgraph.clone()
-    replace_support_point = FiniteLogpPointRewrite()
+    replace_support_point = SupportPointRewrite()
     replace_support_point.rewrite(fgraph)
     # Replace dummy inner inputs by outer inputs
     fgraph.replace_all(tuple(zip(op.inner_inputs, args)), import_missing=True)