chore: use more ruff (#524)

Signed-off-by: Henry Schreiner <[email protected]>

Author: henryiii

.pre-commit-config.yaml

@@ -16,31 +16,13 @@ repos:
   - id: requirements-txt-fixer
   - id: trailing-whitespace
 
-- repo: https://github.com/asottile/setup-cfg-fmt
-  rev: "v2.5.0"
-  hooks:
-  - id: setup-cfg-fmt
-    args: ["--include-version-classifiers", "--max-py-version=3.12"]
-
-- repo: https://github.com/nbQA-dev/nbQA
-  rev: 1.8.7
-  hooks:
-  - id: nbqa-pyupgrade
-    additional_dependencies: [pyupgrade]
-    args: ["--py38-plus"]
-
-- repo: https://github.com/psf/black-pre-commit-mirror
-  rev: 24.8.0
-  hooks:
-  - id: black-jupyter
-
 - repo: https://github.com/astral-sh/ruff-pre-commit
   # Ruff version.
-  rev: v0.6.7
+  rev: v0.6.9
   hooks:
     # Run the linter.
     - id: ruff
-      args: [ --fix ]
+      args: [ --fix, --show-fixes ]
     # Run the formatter.
     - id: ruff-format
 

docs/source/conf.py

@@ -22,11 +22,9 @@
 # Add mplhep to path for sphinx-automodapi
 sys.path.insert(0, os.path.abspath("../../src"))
 
-import mplhep  # noqa: E402
 from pathlib import Path
 
-print("sys.path:", sys.path)
-print("mplhep version:", mplhep.__version__)
+import mplhep
 
 # -- Project information -----------------------------------------------------
 
@@ -70,19 +68,18 @@ def linkcode_resolve(domain, info):
     mod = importlib.import_module(info["module"])
     modpath = [p for p in sys.path if mod.__file__.startswith(p)]
     if len(modpath) < 1:
-        raise RuntimeError("Cannot deduce module path")
+        msg = "Cannot deduce module path"
+        raise RuntimeError(msg)
     modpath = modpath[0]
-    obj = reduce(getattr, [mod] + info["fullname"].split("."))
+    obj = reduce(getattr, [mod, *info["fullname"].split(".")])
     try:
         path = inspect.getsourcefile(obj)
         relpath = path[len(modpath) + 1 :]
         _, lineno = inspect.getsourcelines(obj)
     except TypeError:
         # skip property or other type that inspect doesn't like
         return None
-    return "http://github.com/scikit-hep/mplhep/blob/{}/{}#L{}".format(
-        githash, relpath, lineno
-    )
+    return f"http://github.com/scikit-hep/mplhep/blob/{githash}/{relpath}#L{lineno}"
 
 
 intersphinx_mapping = {

examples/Examples.ipynb

@@ -11,8 +11,8 @@
    },
    "outputs": [],
    "source": [
-    "import numpy as np\n",
     "import matplotlib.pyplot as plt\n",
+    "import numpy as np\n",
     "\n",
     "import mplhep as hep"
    ]

pyproject.toml

@@ -36,9 +36,7 @@ dependencies = [
 
 [project.optional-dependencies]
 dev = [
-    "black",
     "bumpversion",
-    "flake8",
     "jupyter",
     "pre-commit",
     "twine",
@@ -67,14 +65,16 @@ version.source = "vcs"
 build.hooks.vcs.version-file = "src/mplhep/_version.py"
 
 
-[tool.nbqa.mutate]
-pyupgrade = 1
-
-[tool.nbqa.addopts]
-pyupgrade = ["--py38-plus"]
+[tool.uv]
+environments = [
+  "python_version >= '3.10'",
+]
+dev-dependencies = [
+  "mplhep[test]",
+]
 
 
-[tools.mypy]
+[tool.mypy]
 files = ["src"]
 python_version = 3.8
 warn_unused_configs = true
@@ -85,7 +85,7 @@ allow_redefinition = true
 # disallow_untyped_calls = true
 # disallow_untyped_defs = true
 # disallow_incomplete_defs = true
-check_untyped_defs = true
+check_untyped_defs = false
 # disallow_untyped_decorators = true
 # no_implicit_optional = true
 # warn_redundant_casts = true
@@ -106,5 +106,41 @@ testpaths = ["tests"]
 python_files = "test*.py"
 
 
-[tool.isort]
-profile = "black"
+[tool.ruff.lint]
+extend-select = [
+  "B",        # flake8-bugbear
+  "I",        # isort
+  "ARG",      # flake8-unused-arguments
+  "C4",       # flake8-comprehensions
+  "EM",       # flake8-errmsg
+  "ICN",      # flake8-import-conventions
+  "G",        # flake8-logging-format
+  "PGH",      # pygrep-hooks
+  "PIE",      # flake8-pie
+  "PL",       # pylint
+  "PT",       # flake8-pytest-style
+  "RET",      # flake8-return
+  "RUF",      # Ruff-specific
+  "SIM",      # flake8-simplify
+  "T20",      # flake8-print
+  "UP",       # pyupgrade
+  "YTT",      # flake8-2020
+  "EXE",      # flake8-executable
+  "NPY",      # NumPy specific rules
+  "PD",       # pandas-vet
+  "FURB",     # refurb
+  "PYI",      # flake8-pyi
+]
+ignore = [
+  "PLR09",    # Too many <...>
+  "PLR2004",  # Magic value used in comparison
+  "ISC001",   # Conflicts with formatter
+  "NPY002",   # np.random.Generator
+  "G004",     # Logging with f-string
+  "PD011",    # .values vs .to_numpy can get confused
+  "PT013",    # Import from pytest
+]
+
+[tool.ruff.lint.per-file-ignores]
+"tests/**" = ["T20"]
+"docs/conf.py" = ["T20"]

src/mplhep/__init__.py

@@ -23,8 +23,8 @@
     rescale_to_axessize,
     sort_legend,
     ylow,
-    yscale_legend,
     yscale_anchored_text,
+    yscale_legend,
 )
 from .styles import set_style
 

src/mplhep/_deprecate.py

@@ -48,13 +48,9 @@ def __init__(self, name, reason="", warn_once: bool = True, warning=FutureWarnin
 
     def __call__(self, func):
         def decorated_func(*args, **kwargs):
-            if self._name in kwargs.keys() and not (
-                self._warn_once and self._already_warned
-            ):
+            if self._name in kwargs and not (self._warn_once and self._already_warned):
                 warnings.warn(
-                    'kwarg "{}" in function ``{}`` is deprecated and may be removed in future versions: {}'.format(
-                        self._name, func.__name__, self._reason
-                    ),
+                    f'kwarg "{self._name}" in function ``{func.__name__}`` is deprecated and may be removed in future versions: {self._reason}',
                     category=self._warning,
                     stacklevel=2,
                 )

src/mplhep/_version.pyi

@@ -1,4 +1,2 @@
-from __future__ import annotations
-
 version: str
 version_tuple: tuple[int, int, int] | tuple[int, int, int, str, str]

src/mplhep/alice.py

@@ -19,7 +19,7 @@ def text(text="", **kwargs):
     for key, value in dict(mplhep.rcParams.text._get_kwargs()).items():
         if (
             value is not None
-            and key not in kwargs.keys()
+            and key not in kwargs
             and key in inspect.getfullargspec(label_base.exp_text).kwonlyargs
         ):
             kwargs.setdefault(key, value)
@@ -31,7 +31,7 @@ def label(label=None, **kwargs):
     for key, value in dict(mplhep.rcParams.label._get_kwargs()).items():
         if (
             value is not None
-            and key not in kwargs.keys()
+            and key not in kwargs
             and key in inspect.getfullargspec(label_base.exp_label).kwonlyargs
         ):
             kwargs.setdefault(key, value)

src/mplhep/atlas.py

@@ -21,7 +21,7 @@ def text(text="", **kwargs):
     for key, value in dict(mplhep.rcParams.text._get_kwargs()).items():
         if (
             value is not None
-            and key not in kwargs.keys()
+            and key not in kwargs
             and key in inspect.getfullargspec(label_base.exp_text).kwonlyargs
         ):
             kwargs.setdefault(key, value)
@@ -35,7 +35,7 @@ def label(label=None, **kwargs):
     for key, value in dict(mplhep.rcParams.label._get_kwargs()).items():
         if (
             value is not None
-            and key not in kwargs.keys()
+            and key not in kwargs
             and key in inspect.getfullargspec(label_base.exp_label).kwonlyargs
         ):
             kwargs.setdefault(key, value)

src/mplhep/cms.py

@@ -21,7 +21,7 @@ def text(text="", **kwargs):
     for key, value in dict(mplhep.rcParams.text._get_kwargs()).items():
         if (
             value is not None
-            and key not in kwargs.keys()
+            and key not in kwargs
             and key in inspect.getfullargspec(label_base.exp_text).kwonlyargs
         ):
             kwargs.setdefault(key, value)
@@ -35,7 +35,7 @@ def label(label=None, **kwargs):
     for key, value in dict(mplhep.rcParams.label._get_kwargs()).items():
         if (
             value is not None
-            and key not in kwargs.keys()
+            and key not in kwargs
             and key in inspect.getfullargspec(label_base.exp_label).kwonlyargs
         ):
             kwargs.setdefault(key, value)