Skip to content

Commit 2e18aba

Browse files
pre-commit-ci[bot]cspades
authored andcommitted
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 24066b4 commit 2e18aba

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

tests/pytorch/distributed/test_torch_fsdp2.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -134,9 +134,7 @@ def _run_allgather_test(sharding_dims, recipe):
134134
def test_fp8_fsdp2_allgather(sharding_dims, fp_recipe):
135135
"""Verify FSDP2 FP8 all-gather matches a manual dequantize-then-gather reference."""
136136
if fp_recipe == "NVFP4BlockScaling":
137-
pytest.xfail(
138-
f"{fp_recipe}: NVFP4 FSDP2 all-gather hooks need to be implemented."
139-
)
137+
pytest.xfail(f"{fp_recipe}: NVFP4 FSDP2 all-gather hooks need to be implemented.")
140138

141139
parallel_size = math.prod(x for x in sharding_dims if x != 0)
142140
if NUM_PROCS < parallel_size:

0 commit comments

Comments
 (0)