fix: yapf linting

pytorch · Oct 30, 2024 · 29b0edb · 29b0edb
1 parent e34aa0c
commit 29b0edb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_xla/distributed/zero_redundancy_optimizer.py b/torch_xla/distributed/zero_redundancy_optimizer.py
@@ -433,7 +433,7 @@ def allgather_weights_and_update_full_parameter(self, sharding_scheme=None):
 
     # All gather the new weights across the ranks and assign them to the full parameters
     if sharding_scheme is None:
-        sharding_scheme = self._get_sharding_scheme({})
+      sharding_scheme = self._get_sharding_scheme({})
     sharded_data = []
     for param_group, sharded_param_group in zip(
         self.param_groups, self.base_optimizer.param_groups):