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

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

thunder/core/rematerialization.py

@@ -514,6 +514,7 @@ def rematerialize_all_gather(fw_trace: TraceCtx, bw_trace: TraceCtx) -> tuple[Tr
         _update_backward_with_new_saved_for_backward,
         _update_forward_with_new_saved_for_backward,
     )
+
     _update_backward_with_new_saved_for_backward(new_bw_trace, new_required_for_backward)
 
     new_fw_trace = from_trace(fw_trace)

thunder/tests/distributed/test_fsdp.py

@@ -399,7 +399,10 @@ def reset_parameters(self):
         assert model.buf.device.type == "cuda"
 
     # This is not updated yet for joint forward-backward trace
-    @common_utils.decorateIf(unittest.expectedFailure, lambda params: params["bucketing_strategy"] in (FSDPBucketingStrategy.LAYER, FSDPBucketingStrategy.BLOCK))
+    @common_utils.decorateIf(
+        unittest.expectedFailure,
+        lambda params: params["bucketing_strategy"] in (FSDPBucketingStrategy.LAYER, FSDPBucketingStrategy.BLOCK),
+    )
     @common_utils.parametrize(
         "executor,bucketing_strategy,fsdptype",
         product(