Use model context in ValueGradFunction to respect check_bounds

ricardoV94 · ricardoV94 · commit 6e15762f3fe3 · 2025-03-31T18:16:48.000+02:00
diff --git a/pymc/model/core.py b/pymc/model/core.py
@@ -253,7 +253,8 @@ def __init__(
                 )
             inputs = grad_vars
 
-        self._pytensor_function = compile(inputs, outputs, givens=givens, **kwargs)
+        with model:
+            self._pytensor_function = compile(inputs, outputs, givens=givens, **kwargs)
         self._raveled_inputs = ravel_inputs
 
     def set_weights(self, values):

Original file line number	Diff line number	Diff line change
`@@ -253,7 +253,8 @@ def __init__(`
`253`	`253`	`)`
`254`	`254`	`inputs = grad_vars`
`255`	`255`
`256`		`- self._pytensor_function = compile(inputs, outputs, givens=givens, **kwargs)`
	`256`	`+ with model:`
	`257`	`+ self._pytensor_function = compile(inputs, outputs, givens=givens, **kwargs)`
`257`	`258`	`self._raveled_inputs = ravel_inputs`
`258`	`259`
`259`	`260`	`def set_weights(self, values):`