Fix Linting

Author: hsuan-lun-chiang

src/maxtext/layers/nnx_decoders.py

@@ -516,15 +516,15 @@ def _update_leaf(leaf):
           replace_kwargs = {}
           if hasattr(leaf, "get_metadata"):
             replace_kwargs.update(leaf.get_metadata())
-            
+
           replace_kwargs[nnx.PARTITION_NAME] = metadata_axis_name
           replace_kwargs["param_scan_axis"] = axis
-          
+
           for key in ["sharding", "out_sharding", "kernel_axes", "sharding_names"]:
             val = getattr(leaf, key, None)
             if val is None and key in replace_kwargs:
               val = replace_kwargs[key]
-              
+
             if val is not None:
               if isinstance(val, str):
                 val = (val,)