Use ruff over pyink

.github/workflows/CPUTests.yml

@@ -44,13 +44,13 @@ jobs:
     - name: Install Dependencies
       run: |
         python3 -m pip install --upgrade pip
-        python3 -m pip install pylint pyink pytype==2024.2.27
+        python3 -m pip install pylint ruff pytype==2024.2.27
     - name: Typecheck the code with pytype
       run: |
         pytype --jobs auto --disable 'import-error,late-directive,wrong-arg-types,module-attr,unsupported-operands' src/MaxText/ || true
     - name: pylint
       run: |
          pylint --disable=R0401,R0917,W0201,W0613 --ignore-patterns='.pytype,.*pyi$' benchmarks end_to_end src tests
-    - name: pyink
+    - name: ruff
       run: |
-        pyink --pyink-indentation=2 --line-length=122 --check .
+         ruff --config pyproject.toml format --check

.pre-commit-config.yaml

@@ -9,7 +9,7 @@ repos:
         args:
           - '-w'
           - '--skip="*.txt,pylintrc,.*,src/MaxText/assets/*"'
-          - '-L ND,nd,sems,TE,ROUGE,rouge,astroid'
+          - '-L ND,nd,sems,TE,ROUGE,rouge,astroid,ags'
           - '.'
         additional_dependencies:
           - tomli
@@ -44,11 +44,23 @@ repos:
   #          - '--keep-going'
   #          - 'src/MaxText/'
 
-  - repo: https://github.com/google/pyink
-    rev: 24.10.1
+  #  - repo: https://github.com/google/pyink
+  #    rev: 24.10.1
+  #    hooks:
+  #      - id: pyink
+  #        args:
+  #          - '--pyink-indentation=2'
+  #          - '--line-length=122'
+  #          - '--check'
+
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.14.0
     hooks:
-      - id: pyink
+      # Run the linter.
+      # - id: ruff-check
+      # Run the formatter.
+      - id: ruff-format
         args:
-          - '--pyink-indentation=2'
-          - '--line-length=122'
+          - '--config'
+          - 'pyproject.toml'
           - '--check'

.pre-commit-hooks.yaml

@@ -6,14 +6,14 @@
   exclude: '^\.[^/]*(cache|assets|uv|venv)/.*'
   types: [ text ]
 
-- id: pyink
-  name: pyink
-  description: Run pyink Python code formatter.
-  entry: pyink
-  language: python
-  types:
-    - python
-    - text
+#- id: pyink
+#  name: pyink
+#  description: Run pyink Python code formatter.
+#  entry: pyink
+#  language: python
+#  types:
+#    - python
+#    - text
 
 # - id: pytype
 #   name: Pytype
@@ -38,3 +38,11 @@
 #     - toml
 #   files: '^pyproject\.toml$'
 #   entry: validate-pyproject
+
+- id: ruff-format
+  name: ruff-format
+  description: uses `ruff` to prettify code
+  types:
+     - python
+  entry: ruff-format
+  language: system