diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 955040155..f80b1ad1a 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -48,7 +48,7 @@ repos: - --exclude=binder/ - --exclude=versioneer.py - repo: https://github.com/astral-sh/ruff-pre-commit - rev: v0.8.0 + rev: v0.8.1 hooks: - id: ruff args: [--fix, --show-fixes] diff --git a/pymc/step_methods/hmc/nuts.py b/pymc/step_methods/hmc/nuts.py index cc29e0334..770605f4b 100644 --- a/pymc/step_methods/hmc/nuts.py +++ b/pymc/step_methods/hmc/nuts.py @@ -242,24 +242,24 @@ def competence(var, has_grad): class _Tree: __slots__ = ( - "ndim", - "integrator", - "start", - "step_size", "Emax", - "start_energy", - "rng", - "left", - "right", - "proposal", "depth", - "log_size", + "floatX", + "integrator", + "left", "log_accept_sum", + "log_size", + "max_energy_change", "mean_tree_accept", "n_proposals", + "ndim", "p_sum", - "max_energy_change", - "floatX", + "proposal", + "right", + "rng", + "start", + "start_energy", + "step_size", ) def __init__(