Update pre-commit to use ruff

Author: jessegrabowski

.pre-commit-config.yaml

@@ -11,12 +11,6 @@ repos:
         args: [--branch, main]
     -   id: trailing-whitespace
 
-- repo: https://github.com/asottile/pyupgrade
-  rev: v3.16.0
-  hooks:
-    - id: pyupgrade
-      args: [--py310-plus]
-
 - repo: https://github.com/astral-sh/ruff-pre-commit
   rev: v0.4.8
   hooks:

pyproject.toml

@@ -22,20 +22,50 @@ filterwarnings =[
   'ignore:jax\.tree_map is deprecated:DeprecationWarning',
 ]
 
-[tool.black]
-line-length = 100
-
 [tool.coverage.report]
 exclude_lines = [
   "pragma: nocover",
   "raise NotImplementedError",
   "if TYPE_CHECKING:",
 ]
 
-[tool.nbqa.mutate]
-isort = 1
-black = 1
-pyupgrade = 1
+[tool.ruff]
+line-length = 100
+target-version = "py310"
+
+[tool.ruff.lint]
+select = ["D", "E", "F", "I", "UP", "W", "RUF"]
+ignore = [
+  "E501",
+  "RUF001", # String contains ambiguous character (such as Greek letters)
+  "RUF002", # Docstring contains ambiguous character (such as Greek letters)
+  "RUF012", # Mutable class attributes should be annotated with `typing.ClassVar`
+  "D100",
+  "D101",
+  "D102",
+  "D103",
+  "D104",
+  "D105",
+  "D107",
+  "D200",
+  "D202",
+  "D203",
+  "D204",
+  "D205",
+  "D209",
+  "D212",
+  "D213",
+  "D301",
+  "D400",
+  "D401",
+  "D403",
+  "D413",
+  "D415",
+  "D417",
+]
+
+[tool.ruff.lint.isort]
+lines-between-types = 1
 
 [tool.ruff.lint.per-file-ignores]
 'tests/*.py' = ['F841']  # Disable unused variable warning for test files