Skip to content

Commit 38f061c

Browse files
committed
Rename support point rewrite
1 parent 2eaa0e3 commit 38f061c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pymc/distributions/custom.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -399,7 +399,7 @@ def _infer_final_signature(signature: str, n_inputs, n_outputs, n_rngs) -> str:
399399
return signature
400400

401401

402-
class FiniteLogpPointRewrite(GraphRewriter):
402+
class SupportPointRewrite(GraphRewriter):
403403
def rewrite_support_point_scan_node(self, node):
404404
if not isinstance(node.op, Scan):
405405
return
@@ -458,7 +458,7 @@ def dist_support_point(op, rv, *args):
458458
rv_out_idx = node.outputs.index(rv)
459459

460460
fgraph = op.fgraph.clone()
461-
replace_support_point = FiniteLogpPointRewrite()
461+
replace_support_point = SupportPointRewrite()
462462
replace_support_point.rewrite(fgraph)
463463
# Replace dummy inner inputs by outer inputs
464464
fgraph.replace_all(tuple(zip(op.inner_inputs, args)), import_missing=True)

0 commit comments

Comments
 (0)