Skip to content

Commit 68a7578

Browse files
maresbricardoV94
authored andcommitted
Fix test failures due to old change_flags API
Breakage was caused by this: <pymc-devs/pytensor@936554c> Fix is to unpack the dict into kwargs.
1 parent 9b7e759 commit 68a7578

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/sampling/test_mcmc.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -846,7 +846,7 @@ def test_step_vars_in_model(self):
846846
class TestType:
847847
samplers = (Metropolis, Slice, HamiltonianMC, NUTS)
848848

849-
@pytensor.config.change_flags({"floatX": "float64", "warn_float64": "ignore"})
849+
@pytensor.config.change_flags(floatX="float64", warn_float64="ignore")
850850
def test_float64(self):
851851
with pm.Model() as model:
852852
x = pm.Normal("x", initval=np.array(1.0, dtype="float64"))
@@ -861,7 +861,7 @@ def test_float64(self):
861861
warnings.filterwarnings("ignore", ".*number of samples.*", UserWarning)
862862
pm.sample(draws=10, tune=10, chains=1, step=sampler())
863863

864-
@pytensor.config.change_flags({"floatX": "float32", "warn_float64": "warn"})
864+
@pytensor.config.change_flags(floatX="float32", warn_float64="warn")
865865
def test_float32(self):
866866
with pm.Model() as model:
867867
x = pm.Normal("x", initval=np.array(1.0, dtype="float32"))

0 commit comments

Comments
 (0)