Skip to content

Commit e2b73af

Browse files
committed
DEBUG: disable numba cache in test_progressbar_nested_compound
Investigating systematic Bad initial energy failures of this test on the ubuntu (numba, 3.14, ...) CI job. Theory per pytensor#2051: the custom numba cache is not safe across multiprocessing, and a cache collision produces inconsistent compiled logps between the parent process and chain children.
1 parent 83975e3 commit e2b73af

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

tests/progress_bar/test_manager.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414

1515
from unittest.mock import patch
1616

17+
import pytensor
1718
import pytest
1819

1920
import pymc as pm
@@ -153,7 +154,8 @@ def test_smc_multi_chain(imh_kernel):
153154

154155
def test_progressbar_nested_compound():
155156
"""Regression test for https://github.com/pymc-devs/pymc/issues/7721"""
156-
with pm.Model():
157+
# DEBUG: disable numba cache to test cache-race hypothesis for CI flake
158+
with pytensor.config.change_flags(numba__cache=False), pm.Model():
157159
a = pm.Poisson("a", mu=10)
158160
b = pm.Binomial("b", n=a, p=0.8)
159161
c = pm.Poisson("c", mu=11)

0 commit comments

Comments
 (0)