style: apply ruff formatting to FSDP2 changes

2026-07-23 16:30:10 +08:00
parent 4e38183e86
commit 021e6f3788
2 changed files with 4 additions and 4 deletions
@@ -363,9 +363,7 @@ class FSDP2Executor(BaseExecutor):

    @contextmanager
    def _no_sync(self, model: nn.Module):
-        fsdp_modules = [
-            m for m in model.modules() if isinstance(m, FSDPModule)
-        ]
+        fsdp_modules = [m for m in model.modules() if isinstance(m, FSDPModule)]
        if fsdp_modules:
            for m in fsdp_modules:
                m.set_requires_gradient_sync(False, recurse=True)