diff --git a/.flake8 b/.flake8
deleted file mode 100644
index b807b89..0000000
--- a/.flake8
+++ /dev/null
@@ -1,3 +0,0 @@
-[flake8]
-max-line-length = 200
-ignore = F401  # To do: remove this before v4.4
diff --git a/.github/workflows/validate_python.yml b/.github/workflows/validate_python.yml
index 896a2a8..331fc03 100644
--- a/.github/workflows/validate_python.yml
+++ b/.github/workflows/validate_python.yml
@@ -12,22 +12,19 @@ on:
   workflow_dispatch:
 
 jobs:
-  flake8_py3:
+  flake8-lint:
     runs-on: ubuntu-latest
+    name: Lint
     steps:
-      - name: Setup Python
-        uses: actions/setup-python@v1
+      - name: Check out source repository
+        uses: actions/checkout@v3
+      - name: Set up Python environment
+        uses: actions/setup-python@v4
         with:
-          python-version: 3.7.17
-          architecture: x64
-      - name: Checkout PyTorch
-        uses: actions/checkout@main
-      - name: Install flake8
-        run: pip install flake8
-      - name: Run flake8
-        uses: suo/flake8-github-action@releases/v1
+          python-version: "3.11"
+      - name: flake8 Lint
+        uses: py-actions/flake8@v2
         with:
-          checkName: 'flake8_py3'   # NOTE: this needs to be the same as the job name
-        env:
-          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          ignore: "F401"
+          max-line-length: "200"
 ...