chore: specifiy ruff version

Author: wooway777

.github/workflows/build.yml

@@ -27,7 +27,7 @@ jobs:
       if: matrix.os == 'ubuntu-latest' && matrix.type == 'debug'
       run: |
         sudo apt-get install -y clang-format-16
-        pip install black ruff colorama
+        pip install black ruff==0.15.20 colorama
         if [ "${{ github.event_name }}" = "pull_request" ]; then
           python3 scripts/format.py --ref "${{ github.event.pull_request.base.sha }}" --path src --check
         elif [ "${{ github.event.before }}" != "0000000000000000000000000000000000000000" ]; then

.github/workflows/ruff.yml

@@ -8,7 +8,9 @@ jobs:
       - uses: chartboost/ruff-action@v1
         with:
           src: './python/'
+          version: "0.15.20"
       - uses: chartboost/ruff-action@v1
         with:
           src: './python/'
           args: format --check
+          version: "0.15.20"

python/infinicore/nn/functional/layer_norm.py

@@ -15,9 +15,9 @@ def layer_norm(
 ) -> Tensor:
     r"""Apply Layer Normalization."""
 
-    assert (
-        normalized_shape == weight.shape
-    ), "normalized_shape does not match weight.shape."
+    assert normalized_shape == weight.shape, (
+        "normalized_shape does not match weight.shape."
+    )
 
     if out is None:
         return Tensor(

python/infinicore/nn/functional/rms_norm.py

@@ -14,9 +14,9 @@ def rms_norm(
 ) -> Tensor:
     r"""Apply Root Mean Square Layer Normalization."""
 
-    assert (
-        normalized_shape == weight.shape
-    ), "normalized_shape does not match weight.shape."
+    assert normalized_shape == weight.shape, (
+        "normalized_shape does not match weight.shape."
+    )
 
     if out is None:
         return Tensor(_infinicore.rms_norm(input._underlying, weight._underlying, eps))

scripts/format.py

@@ -78,8 +78,9 @@ def format_file(file: Path, check: bool, formatter) -> bool:
                     check=True,
                 )
                 print(f"{Fore.CYAN}Formatted: {file}{Style.RESET_ALL}")
-
+        elif formatter == "ruff":
             ruff_cmd = ["ruff", "check", file]
+            ruff_format_cmd = ["ruff", "format", file]
             try:
                 if check:
                     process = subprocess.run(
@@ -94,6 +95,22 @@ def format_file(file: Path, check: bool, formatter) -> bool:
                             f"Use {Fore.CYAN}ruff check --fix {file}{Style.RESET_ALL} to fix it."
                         )
                         formatted = False
+
+                    ruff_format_cmd.insert(2, "--check")
+                    process = subprocess.run(
+                        ruff_format_cmd,
+                        capture_output=True,
+                        text=True,
+                        check=False,
+                    )
+                    if process.returncode != 0:
+                        print(
+                            f"{Fore.YELLOW}{file} is not ruff-formatted.{Style.RESET_ALL}"
+                        )
+                        print(
+                            f"Use {Fore.CYAN}ruff format {file}{Style.RESET_ALL} to format it."
+                        )
+                        formatted = False
                 else:
                     ruff_cmd.insert(2, "--fix")
                     process = subprocess.run(
@@ -111,6 +128,24 @@ def format_file(file: Path, check: bool, formatter) -> bool:
                         formatted = False
                     else:
                         print(f"{Fore.CYAN}Ruff fixed: {file}{Style.RESET_ALL}")
+
+                    process = subprocess.run(
+                        ruff_format_cmd,
+                        capture_output=True,
+                        text=True,
+                        check=False,
+                    )
+                    if process.returncode != 0:
+                        print(
+                            f"{Fore.RED}Ruff format failed for {file}.{Style.RESET_ALL}"
+                        )
+                        if process.stdout:
+                            print(process.stdout, end="")
+                        if process.stderr:
+                            print(process.stderr, end="")
+                        formatted = False
+                    else:
+                        print(f"{Fore.CYAN}Ruff formatted: {file}{Style.RESET_ALL}")
             except FileNotFoundError:
                 print(
                     f"{Fore.RED}Formatter ruff not found, {file} skipped.{Style.RESET_ALL}"
@@ -217,9 +252,7 @@ def main():
     parser.add_argument(
         "--c", default="clang-format-16", help="C formatter (default: clang-format-16)"
     )
-    parser.add_argument(
-        "--py", default="black", help="Python formatter (default: black)"
-    )
+    parser.add_argument("--py", default="ruff", help="Python formatter (default: ruff)")
     args = parser.parse_args()
 
     if args.ref is None and args.path is None: