Improve current ruff rules (#1720)

* improve current ruff rules

* rm comments

* fix lengths

* trim

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()

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"