Update torch-ext/optimizer/muon.py

torch-ext/optimizer/muon.py

@@ -656,7 +656,7 @@ class Muon(torch.optim.Optimizer):
         for n, p in zip(ordered_names, ordered_params):
             if mesh is None:
                 mesh = p.device_mesh
-                shard_mesh, process_group = self.get_shard_mesh(p)                
+                shard_mesh, process_group = self.get_shard_mesh(p)
             elif mesh != p.device_mesh:
                 raise ValueError("All parameters must be on the same mesh.")
             num_ranks = dist.get_world_size(group=process_group)