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

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

Author: pre-commit-ci[bot]

tests/jax/test_custom_call_compute.py

@@ -682,9 +682,7 @@ def test_grouped_qdq(
         x = jax.random.uniform(subkeys[1], input_shape, in_dtype)
 
         grouped_quantizer = QuantizerFactory.create(
-            q_params=QuantizerParams(
-                q_dtype=q_dtype, scaling_mode=scaling_mode, q_layout=q_layout
-            ),
+            q_params=QuantizerParams(q_dtype=q_dtype, scaling_mode=scaling_mode, q_layout=q_layout),
             n_groups=n_groups,
         )
 

transformer_engine/jax/quantize/quantizer.py

@@ -804,15 +804,21 @@ def create(
         Returns:
             A single quantizer or tuple of quantizers
         """
-        assert all(arg_name not in kwargs for arg_name in ['scaling_mode', 'q_layout', 'q_dtype']), "Please use q_params instead of passing scaling_mode, q_layout, and q_dtype as separate args."
+        assert all(
+            arg_name not in kwargs for arg_name in ["scaling_mode", "q_layout", "q_dtype"]
+        ), (
+            "Please use q_params instead of passing scaling_mode, q_layout, and q_dtype as separate"
+            " args."
+        )
         assert isinstance(q_params.scaling_mode, ScalingMode), "Invalid scaling_mode type"
         if q_params.scaling_mode == ScalingMode.NO_SCALING:
             quantizers = [None] * n_quantizers
         else:
             if n_groups:
                 if n_quantizers != 1:
                     warnings.warn(
-                        "Using more than one GroupedQuantizer for a grouped input is not recommended"
+                        "Using more than one GroupedQuantizer for a grouped input is not"
+                        " recommended"
                     )
                 quantizer_type = GroupedQuantizer
                 kwargs["n_groups"] = n_groups