From 692236a6eaedd72ed3e307fe4eaac9a7fb292d9d Mon Sep 17 00:00:00 2001
From: Driss Guessous <32754868+drisspg@users.noreply.github.com>
Date: Thu, 19 Dec 2024 17:02:50 -0800
Subject: [PATCH] Add ruff config options in actiosn to ruff toml (#1453)

stack-info: PR: https://github.com/pytorch/ao/pull/1453, branch: drisspg/stack/23
---
 .github/workflows/ruff_linter.yml | 24 ++++++++++++------------
 .pre-commit-config.yaml           |  6 ++++++
 ruff.toml                         |  1 +
 3 files changed, 19 insertions(+), 12 deletions(-)

diff --git a/.github/workflows/ruff_linter.yml b/.github/workflows/ruff_linter.yml
index 027279721e..40ac883e71 100644
--- a/.github/workflows/ruff_linter.yml
+++ b/.github/workflows/ruff_linter.yml
@@ -22,7 +22,7 @@ jobs:
     permissions:
       contents: write
       pull-requests: write
-    
+
     strategy:
       matrix:
         python-version: ["3.9"]
@@ -33,43 +33,43 @@ jobs:
         PR_URL=${{ github.event.inputs.pr_url }}
         PR_NUMBER=$(echo $PR_URL | grep -oE '[0-9]+$')
         echo "PR_NUMBER=$PR_NUMBER" >> $GITHUB_ENV
-        
+
     - uses: actions/checkout@v3
       if: github.event_name == 'workflow_dispatch'
       with:
         fetch-depth: 0
         token: ${{ secrets.GITHUB_TOKEN }}
-    
+
     - name: Checkout PR branch
       if: github.event_name == 'workflow_dispatch'
       run: |
         gh pr checkout ${{ env.PR_NUMBER }}
       env:
         GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
-        
+
     - uses: actions/checkout@v3
       if: github.event_name != 'workflow_dispatch'
       with:
         fetch-depth: 0
-        
+
     - name: Set up Python ${{ matrix.python-version }}
       uses: actions/setup-python@v3
       with:
         python-version: ${{ matrix.python-version }}
-        
+
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
         pip install ruff==0.6.8
-        
+
     - name: Regular lint check
       if: github.event_name != 'workflow_dispatch'
       run: |
-        ruff check .
         # --isolated is used to skip the allowlist at all so this applies to all files
         # please be careful when using this large changes means everyone needs to rebase
+        # if you do be sure to update .pre-commit-config.yaml
         ruff check --isolated --select F821,F823,W191
-        ruff check --select F,I
+        ruff check
         ruff format --check || {
           echo "Ruff check failed, please try again after running 'ruff format'."
           exit 1
@@ -80,11 +80,11 @@ jobs:
       run: |
         git config --global user.name 'github-actions[bot]'
         git config --global user.email 'github-actions[bot]@users.noreply.github.com'
-        
+
         # Apply fixes
-        ruff check --select F,I --fix
+        ruff check --fix
         ruff format .
-        
+
         # Commit and push if there are changes
         if [[ -n "$(git status --porcelain)" ]]; then
           git add .
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index d8c7ef4adf..2ac354e6d7 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -21,3 +21,9 @@ repos:
           - F,I
       # Run the formatter.
       - id: ruff-format
+      # Run isolated checks.
+      - id: ruff
+        alias: ruff-isolated
+        args:
+          - --isolated
+          - select F821,F823,W191
diff --git a/ruff.toml b/ruff.toml
index b20cab030c..763a9161e2 100644
--- a/ruff.toml
+++ b/ruff.toml
@@ -20,4 +20,5 @@ include = [
     "test/prototype/low_bit_optim/**.py",
 ]
 
+lint.select = ["F", "I"]
 lint.ignore = ["E731"]