Update ruff version to 0.11.6 (#2103)

Summary:
Use more recent versions so people can have less problems when they just install the latest ruff

Test Plan:
CI

Reviewers:

Subscribers:

Tasks:

Tags:

Author: jerryzh168

.github/scripts/github_utils.py

@@ -50,10 +50,10 @@ def gh_fetch_url_and_headers(
         ):
             print(
                 f"""Rate limit exceeded:
-                Used: {err.headers['X-RateLimit-Used']}
-                Limit: {err.headers['X-RateLimit-Limit']}
-                Remaining: {err.headers['X-RateLimit-Remaining']}
-                Resets at: {err.headers['x-RateLimit-Reset']}"""
+                Used: {err.headers["X-RateLimit-Used"]}
+                Limit: {err.headers["X-RateLimit-Limit"]}
+                Remaining: {err.headers["X-RateLimit-Remaining"]}
+                Resets at: {err.headers["x-RateLimit-Reset"]}"""
             )
         raise
 

.github/scripts/label_utils.py

@@ -62,9 +62,9 @@ def gh_get_labels(org: str, repo: str) -> List[str]:
     update_labels(labels, info)
 
     last_page = get_last_page_num_from_header(header)
-    assert (
-        last_page > 0
-    ), "Error reading header info to determine total number of pages of labels"
+    assert last_page > 0, (
+        "Error reading header info to determine total number of pages of labels"
+    )
     for page_number in range(2, last_page + 1):  # skip page 1
         _, info = request_for_labels(prefix + f"&page={page_number}")
         update_labels(labels, info)

.github/scripts/trymerge.py

@@ -490,7 +490,7 @@ def get_check_run_name_prefix(workflow_run: Any) -> str:
     if workflow_run is None:
         return ""
     else:
-        return f'{workflow_run["workflow"]["name"]} / '
+        return f"{workflow_run['workflow']['name']} / "
 
 
 def is_passing_status(status: Optional[str]) -> bool:
@@ -538,7 +538,7 @@ def add_conclusions(edges: Any) -> None:
                     if not isinstance(checkrun_node, dict):
                         warn(f"Expected dictionary, but got {type(checkrun_node)}")
                         continue
-                    checkrun_name = f'{get_check_run_name_prefix(workflow_run)}{checkrun_node["name"]}'
+                    checkrun_name = f"{get_check_run_name_prefix(workflow_run)}{checkrun_node['name']}"
                     existing_checkrun = workflow_obj.jobs.get(checkrun_name)
                     if existing_checkrun is None or not is_passing_status(
                         existing_checkrun.status
@@ -653,7 +653,7 @@ def skip_func(idx: int, candidate: "GitHubPR") -> bool:
         if not open_only or not candidate.is_closed():
             return False
         print(
-            f"Skipping {idx+1} of {len(rev_list)} PR (#{candidate.pr_num}) as its already been merged"
+            f"Skipping {idx + 1} of {len(rev_list)} PR (#{candidate.pr_num}) as its already been merged"
         )
         return True
 

.github/scripts/trymerge_explainer.py

@@ -77,7 +77,7 @@ def get_merge_message(
             (
                 "<details><summary>Advanced Debugging</summary>",
                 "Check the merge workflow status ",
-                f"<a href=\"{os.getenv('GH_RUN_URL')}\">here</a>",
+                f'<a href="{os.getenv("GH_RUN_URL")}">here</a>',
                 "</details>",
             )
         )

.github/workflows/ruff_linter.yml

@@ -60,7 +60,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install ruff==0.6.8
+        pip install ruff==0.11.6
 
     - name: Regular lint check
       if: github.event_name != 'workflow_dispatch'

benchmarks/bench_galore_fused_kernels.py

@@ -21,7 +21,7 @@ def run(args):
 
     benchmark = get_benchmark(M, N, dtype, allow_tf32=allow_tf32)
     save_path = (
-        f'benchmark_{M}x{N}_{rank}_{args.dtype}_{"tf32" if allow_tf32 else "no-tf32"}'
+        f"benchmark_{M}x{N}_{rank}_{args.dtype}_{'tf32' if allow_tf32 else 'no-tf32'}"
     )
     if not os.path.exists(save_path):
         os.makedirs(save_path)

benchmarks/benchmark_low_bit_adam.py

@@ -185,12 +185,12 @@ def evaluate_model(model, args):
     if args.full_bf16:
         assert args.amp == "none", "When --full_bf16 is set, --amp must be none"
     if args.optim_cpu_offload == "deepspeed":
-        assert (
-            args.amp == "none"
-        ), "When using DeepSpeed ZeRO-Offload, --amp must be none"
-        assert (
-            args.optim == "AdamW"
-        ), "When using DeepSpeed ZeRO-Offload, --optim must be AdamW"
+        assert args.amp == "none", (
+            "When using DeepSpeed ZeRO-Offload, --amp must be none"
+        )
+        assert args.optim == "AdamW", (
+            "When using DeepSpeed ZeRO-Offload, --optim must be AdamW"
+        )
     if args.profile:
         args.n_epochs = 1
     if args.seed is not None:

benchmarks/benchmark_uintx.py

@@ -109,7 +109,7 @@ def uintx_vs_fp16(nbits=[1, 2, 3, 4, 5, 6, 7], scales=[256, 512, 1024], repeats=
     for result in results:
         print(f"scale: {result[0]} fp16 time:{result[1]: .2f}ms speedups:")
         for i in range(2, len(result)):
-            print(f"int{nbits[i-2]}: {result[1]/result[i]: .2f}x")
+            print(f"int{nbits[i - 2]}: {result[1] / result[i]: .2f}x")
 
 
 if __name__ == "__main__":

benchmarks/float8/float8_roofline.py

@@ -225,9 +225,9 @@ def run(
     * `enable_fusion_modeling`: if False uses Linear, if True uses LNLinearSigmoid and models the fusion of float8 overhead
     """
 
-    assert not (
-        (float8_recipe_name is not None) and (mx_recipe_name is not None)
-    ), "unsupported"
+    assert not ((float8_recipe_name is not None) and (mx_recipe_name is not None)), (
+        "unsupported"
+    )
     if float8_recipe_name is None and mx_recipe_name is None:
         float8_recipe_name = "tensorwise"
 

benchmarks/float8/profile_lowp_training.py

@@ -299,22 +299,21 @@ def main(
         "lowp",
         "ref",
     ), "experiment_filter must be one of `both`, `lowp`, `ref`"
-    assert (
-        mode_filter
-        in (
-            "fwd_bwd",
-            "fwd",
-            "cast_only",
-            "cast_with_to_blocked",
-            "cast_only_dim0_dim1",
-        )
-    ), "mode_filter must be one of `fwd_bwd`, `fwd`, `cast_only`, `cast_with_to_blocked`, `cast_only_dim0_dim1`"
+    assert mode_filter in (
+        "fwd_bwd",
+        "fwd",
+        "cast_only",
+        "cast_with_to_blocked",
+        "cast_only_dim0_dim1",
+    ), (
+        "mode_filter must be one of `fwd_bwd`, `fwd`, `cast_only`, `cast_with_to_blocked`, `cast_only_dim0_dim1`"
+    )
     if mode_filter == "cast_only":
         assert experiment_filter == "lowp", "unsupported"
 
-    assert not (
-        float8_recipe_name is not None and mx_recipe_name is not None
-    ), "either float8_recipe_name or mx_recipe_name can be specified, but not both"
+    assert not (float8_recipe_name is not None and mx_recipe_name is not None), (
+        "either float8_recipe_name or mx_recipe_name can be specified, but not both"
+    )
 
     if float8_recipe_name is None and mx_recipe_name is None:
         config = Float8LinearConfig()