Update torch-ext/optimizer/muon.py

torch-ext/optimizer/muon.py

@@ -610,7 +610,7 @@ class Muon(torch.optim.Optimizer):
         elif p.placements == (Replicate(), Shard(dim=0)):
             # Case for HSDP
             process_group = p.device_mesh.get_group(mesh_dim=1)
-            if self.rank is None:           
+            if self.rank is None:
                 self.rank = dist.get_rank(group=process_group)
             else:
                 assert self.rank == dist.get_rank(group=process_group)