Update moe.py

Author: BirdsOfAFthr

src/maxtext/layers/moe.py

@@ -897,7 +897,8 @@ def gmm(
         inputs, kernel, tiling, group_sizes, expert_assignments, weight_gather_axes, input_buffer_count, combine_scopes
     ):
       # TODO (b/491979205) pipeline fsdp ag per repeat fails tokamax gmm
-      if self.config.use_qwix_quantization or (self.config.using_pipeline_parallelism and self.config.pipeline_fsdp_ag_per_repeat):
+      if self.config.use_qwix_quantization or (self.config.using_pipeline_parallelism and 
+                                               self.config.pipeline_fsdp_ag_per_repeat):
         tokamax_group_sizes = group_sizes
       else:
         tokamax_group_sizes = tokamax.RaggedDotGroupSizes(