[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

tests/pytorch/distributed/run_fsdp2_model.py

@@ -560,9 +560,9 @@ def _train(args):
         v_pt = s_post_train[key]
         if isinstance(v_pt, DTensor):
             v_pt = v_pt.to_local()
-        assert not torch.allclose(v1, v_pt), (
-            f"[{key}] Model weights should have changed after extra training steps"
-        )
+        assert not torch.allclose(
+            v1, v_pt
+        ), f"[{key}] Model weights should have changed after extra training steps"
 
     # Load the checkpoint.
     state_dict = {"app": AppState(model=model, optimizer=optimizer)}