From 593525048c58a9234130bff25525be68a0499316 Mon Sep 17 00:00:00 2001 From: Ricardo Vieira Date: Mon, 24 Jun 2024 10:36:53 +0200 Subject: [PATCH] Rename support point rewrite --- pymc/distributions/custom.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pymc/distributions/custom.py b/pymc/distributions/custom.py index 681e4fa16dc..e2ee932e54c 100644 --- 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)