Skip to content

Commit 5a47c8d

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 45ff5fd commit 5a47c8d

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

pymc/model/transform/conditioning.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -359,7 +359,7 @@ def remove_value_transforms(
359359
"""
360360
if vars is None:
361361
vars = model.free_RVs
362-
return change_value_transforms(model, {var: None for var in vars})
362+
return change_value_transforms(model, dict.fromkeys(vars))
363363

364364

365365
__all__ = (

tests/logprob/test_transform_value.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -606,7 +606,7 @@ def _growth(limit=10, peak_stats={}):
606606
return [(name, stats[name], delta) for name, delta in deltas]
607607

608608
rvs_to_values = {pt.random.beta(1, 1, name=f"p_{i}"): pt.scalar(f"p_{i}") for i in range(30)}
609-
tr = TransformValuesRewrite({v: logodds for v in rvs_to_values.values()})
609+
tr = TransformValuesRewrite(dict.fromkeys(rvs_to_values.values(), logodds))
610610

611611
for i in range(20):
612612
conditional_logp(rvs_to_values, extra_rewrites=tr)

0 commit comments

Comments
 (0)