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

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

transformer_engine/jax/flax/transformer.py

@@ -633,7 +633,7 @@ def __post_init__(self):
     def _assert_dtypes(self, query: Array, key: Array, value: Array, qkv_layout: QKVLayout):
         """Asserts that the dtypes of query, key, and value dtypes are consistent."""
         if qkv_layout.is_qkvpacked():
-            pass # No need to check dtypes for key and value since it is packed
+            pass  # No need to check dtypes for key and value since it is packed
         elif qkv_layout.is_kvpacked():
             assert (
                 key.dtype == query.dtype