diff --git a/numpyro/infer/mcmc.py b/numpyro/infer/mcmc.py index 650072845..b4ac1ec4a 100644 --- a/numpyro/infer/mcmc.py +++ b/numpyro/infer/mcmc.py @@ -723,10 +723,10 @@ def print_summary(self, prob=0.9, exclude_deterministic=True): def transfer_states_to_host(self): """ - Reduce the memory footprint of collected samples by transfering them to the host device. + Reduce the memory footprint of collected samples by transfering them to the host device. """ self._states = device_get(self._states) - self._states_flat = device_get(self._states_flat) + self._states_flat = device_get(self._states_flat) def __getstate__(self): state = self.__dict__.copy() diff --git a/pyproject.toml b/pyproject.toml index de1df6bc0..63c7494a5 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -35,9 +35,7 @@ line-length = 88 indent-width = 4 [tool.ruff.lint] -# Enable Pyflakes (`F`) and a subset of the pycodestyle (`E`) codes by default. -# We also add isort. -select = ["E4", "E7", "E9", "F", "I"] +select = ["E", "F", "I", "W"] ignore = ["E203"] # Allow fix for all enabled rules (when `--fix`) is provided. @@ -47,6 +45,9 @@ unfixable = [] # Allow unused variables when underscore-prefixed. dummy-variable-rgx = "^(_+|(_+[a-zA-Z0-9_]*[a-zA-Z0-9]+?))$" +[tool.ruff.lint.pycodestyle] +max-line-length = 120 + [tool.ruff.format] # Like Black, use double quotes for strings. quote-style = "double"