fix: allgather striped window_size (5,0) id says (8,0)

[andrewwhitecdw]

This PR addresses the following issue in tests/jax/test_distributed_fused_attn.py: allgather striped window_size (5,0) id says (8,0).

Changes

• tests/jax/test_distributed_fused_attn.py: allgather striped window_size (5,0) id says (8,0).

Details

--- a/tests/jax/test_distributed_fused_attn.py
+++ b/tests/jax/test_distributed_fused_attn.py
@@ -1,7 +1,7 @@
-    @pytest.mark.parametrize(
-        "window_size",
-        [
-            pytest.param((-1, -1), id="window_size(-1, -1)"),
-            pytest.param((5, 0), id="window_size(8, 0)"),
-        ],
-    )
+    @pytest.mark.parametrize(
+        "window_size",
+        [
+            pytest.param((-1, -1), id="window_size(-1, -1)"),
+            pytest.param((5, 0), id="window_size(5, 0)"),
+        ],
+    )

Tests

Let me know if you want tests added for this fix or not.

[greptile-apps]

Greptile Summary

Corrects the pytest identifier for the striped all-gather fused-attention case so the displayed window size matches the actual (5, 0) parameter.

• Changes only the test-case display identifier.
• Leaves the parameter value and test behavior unchanged.

Confidence Score: 5/5

The PR appears safe to merge because it only corrects test-reporting metadata to match the existing parameter value.

The changed pytest ID now accurately describes the (5, 0) input, with no effect on test selection, inputs, assertions, or runtime behavior.

Important Files Changed

Filename	Overview
tests/jax/test_distributed_fused_attn.py	Correctly updates a mismatched pytest parameter ID from window_size(8, 0) to window_size(5, 0) without changing test execution.

_{Reviews (1): Last reviewed commit: "fix: allgather striped window_size (5,0)..." | Re-trigger Greptile}