[Lint] Migrate from flake8/isort to ruff for faster linting

.github/workflows/lint.yaml

@@ -27,6 +27,6 @@ jobs:
           pre-commit install-hooks
       - name: Get changed files
         id: changed-files
-        uses: tj-actions/changed-files@v46.0.5
+        uses: tj-actions/changed-files@v47.0.0
       - name: Lint modified files
         run: pre-commit run --files ${{ steps.changed-files.outputs.all_changed_files }}

.github/workflows/reusable-ci-tests.yml

@@ -190,7 +190,7 @@ jobs:
       - name: Get changed files
         if: steps.check_skip.outputs.skip_tests == 'false'
         id: changed-files
-        uses: tj-actions/changed-files@v46.0.5
+        uses: tj-actions/changed-files@v47.0.0
 
       # =================================================================
       # STAGE 2: OPS TESTS
@@ -301,7 +301,7 @@ jobs:
       - name: Get changed files
         if: steps.check_skip.outputs.skip_tests == 'false'
         id: changed-files
-        uses: tj-actions/changed-files@v46.0.5
+        uses: tj-actions/changed-files@v47.0.0
 
       - name: Install/Update Dependencies
         if: steps.check_skip.outputs.skip_tests == 'false'

.pre-commit-config.yaml

@@ -1,24 +1,18 @@
 repos:
 - repo: https://github.com/pre-commit/pre-commit-hooks
-  rev: v5.0.0
+  rev: v6.0.0
   hooks:
   - id: check-symlinks
   - id: trailing-whitespace
     args: [--markdown-linebreak-ext=md]
   - id: end-of-file-fixer
   - id: check-yaml
   - id: check-toml
-  - id: check-ast
   - id: check-added-large-files
   - id: check-merge-conflict
   - id: detect-private-key
-  - id: debug-statements
-- repo: https://github.com/pycqa/isort
-  rev: 5.12.0
+- repo: https://github.com/astral-sh/ruff-pre-commit
+  rev: v0.14.0
   hooks:
-  - id: isort
-- repo: https://github.com/PyCQA/flake8
-  rev: 7.0.0
-  hooks:
-  - id: flake8
-    args: [--max-line-length=127]
+  - id: ruff-check
+    args: [--fix, --exit-non-zero-on-fix]

pyproject.toml

@@ -34,13 +34,46 @@ Repository = "https://github.com/fla-org/flash-linear-attention"
 [build-system]
 requires = ["setuptools>=45", "wheel"]
 
-[tool.isort]
-line_length = 127
-multi_line_output = 3
+[tool.ruff.lint.isort]
+known-first-party = ["fla"]
+force-sort-within-sections = false
 
 [tool.pytest.ini_options]
 log_cli = true
 log_cli_level = "INFO"
 pythonpath = [
   "."
 ]
+
+[tool.ruff]
+target-version = "py310"
+line-length = 127
+
+[tool.ruff.format]
+docstring-code-format = true
+
+[tool.ruff.lint]
+select = [
+    "E",    # pycodestyle
+    "F",    # Pyflakes
+    "UP",   # pyupgrade
+    "B",    # flake8-bugbear
+    "SIM",  # flake8-simplify
+    "I",    # isort
+    "C4",   # flake8-comprehensions
+    "TCH",  # flake8-type-checking
+    "COM",  # flake8-commas
+    "T",    # flake8-debugger
+]
+ignore = [
+    "E501",
+    "E741",
+    "B023",
+    "B008",
+    "SIM108",
+]
+
+[tool.ruff.lint.per-file-ignores]
+"__init__.py" = ["F401"]
+"fla/utils.py" = ["TCH004"]
+"evals/harness.py" = ["I", "TCH"]