Do not garbage collect logp_dlogp function

ricardoV94 · ricardoV94 · commit 8478e2fb8f9d · 2025-04-07T16:25:03.000+02:00
diff --git a/pymc/step_methods/hmc/integration.py b/pymc/step_methods/hmc/integration.py
@@ -42,6 +42,7 @@ def __init__(self, potential: QuadPotential, logp_dlogp_func):
         self._potential = potential
         # Sidestep logp_dlogp_function.__call__
         pytensor_function = logp_dlogp_func._pytensor_function
+        pytensor_function.vm.allow_gc = False
         # Create some wrappers for backwards compatibility during transition
         # When raveled_inputs=False is forbidden, func = pytensor_function
         if logp_dlogp_func._raveled_inputs:
