fix linting

AlibabaPAI · Sep 2, 2024 · 9452202 · 9452202
1 parent 8968543
commit 9452202
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/torch_xla/experimental/spmd_fully_sharded_data_parallel.py b/torch_xla/experimental/spmd_fully_sharded_data_parallel.py
@@ -97,11 +97,10 @@ def __init__(
       self._auto_wrap(auto_wrap_kwargs, fsdp_kwargs)
 
     _materialize_module(
-      module,
-      None,
-      [],
-      deferred_init_check_fn=lambda k: not isinstance(
-          k, SpmdFullyShardedDataParallel))
+        module,
+        None, [],
+        deferred_init_check_fn=lambda k: not isinstance(
+            k, SpmdFullyShardedDataParallel))
 
     # Let's move the module to xla device in case it's not moved
     # by the caller already.