Update

hsuan-lun-chiang · hsuan-lun-chiang · commit 8bacbe073918 · 2026-03-26T07:45:11.000Z
diff --git a/src/maxtext/layers/quantizations.py b/src/maxtext/layers/quantizations.py
@@ -149,7 +149,7 @@ def _get_rhs_axis_metadata_wrapper(
     )
 
   def dot_general_cls(self, mesh_axes: Tuple[str, ...] = ()):
-    """Returns dot_general configured with aqt params."""
+    """Returns dot_general configured with aqt params"""
     if isinstance(self.quant_dg, dict):
       quant_dg, is_tiled, tiling_fn = self._get_mixed_precision_cfg()
     else:

Original file line number	Diff line number	Diff line change
`@@ -149,7 +149,7 @@ def _get_rhs_axis_metadata_wrapper(`
`149`	`149`	`)`
`150`	`150`
`151`	`151`	`def dot_general_cls(self, mesh_axes: Tuple[str, ...] = ()):`
`152`		`- """Returns dot_general configured with aqt params."""`
	`152`	`+ """Returns dot_general configured with aqt params"""`
`153`	`153`	`if isinstance(self.quant_dg, dict):`
`154`	`154`	`quant_dg, is_tiled, tiling_fn = self._get_mixed_precision_cfg()`
`155`	`155`	`else:`