[pre-commit.ci] auto fixes from pre-commit.com hooks

pre-commit-ci[bot] · pre-commit-ci[bot] · commit 783ac5f0bc6d · 2026-05-18T09:28:22.000Z
for more information, see https://pre-commit.ci
diff --git a/transformer_engine/pytorch/module/generalized_tensor_parallelism.py b/transformer_engine/pytorch/module/generalized_tensor_parallelism.py
@@ -1462,7 +1462,7 @@ def _buf_bytes(shape, dtype) -> int:
         if dtype not in GTPWeightCache._BYTES_PER_ELEMENT:
             raise KeyError(
                 f"GTPWeightCache._buf_bytes: unknown dtype {dtype!r}.  "
-                f"Add it to GTPWeightCache._BYTES_PER_ELEMENT with its bytes-per-element."
+                "Add it to GTPWeightCache._BYTES_PER_ELEMENT with its bytes-per-element."
             )
         return int(numel * GTPWeightCache._BYTES_PER_ELEMENT[dtype])
 

Original file line number	Diff line number	Diff line change
`@@ -1462,7 +1462,7 @@ def _buf_bytes(shape, dtype) -> int:`
`1462`	`1462`	`if dtype not in GTPWeightCache._BYTES_PER_ELEMENT:`
`1463`	`1463`	`raise KeyError(`
`1464`	`1464`	`f"GTPWeightCache._buf_bytes: unknown dtype {dtype!r}. "`
`1465`		`- f"Add it to GTPWeightCache._BYTES_PER_ELEMENT with its bytes-per-element."`
	`1465`	`+ "Add it to GTPWeightCache._BYTES_PER_ELEMENT with its bytes-per-element."`
`1466`	`1466`	`)`
`1467`	`1467`	`return int(numel * GTPWeightCache._BYTES_PER_ELEMENT[dtype])`
`1468`	`1468`