ruff reformat

Author: pstjohn

src/accelerate/test_utils/testing.py

@@ -543,10 +543,13 @@ def require_transformer_engine(test_case):
 
 def require_transformer_engine_mxfp8(test_case):
     """
-    Decorator marking a test that requires transformers engine FP8 block scaling available. These tests are skipped when transformers
-    engine FP8 block scaling isn't available
+    Decorator marking a test that requires transformers engine FP8 block scaling available. These tests are skipped
+    when transformers engine FP8 block scaling isn't available
     """
-    return unittest.skipUnless(is_transformer_engine_mxfp8_available(), "test requires transformers engine FP8 block scaling")(test_case)
+    return unittest.skipUnless(
+        is_transformer_engine_mxfp8_available(), "test requires transformers engine FP8 block scaling"
+    )(test_case)
+
 
 def require_torchao(test_case):
     """

src/accelerate/utils/imports.py

@@ -117,9 +117,11 @@ def is_transformer_engine_available():
 def is_transformer_engine_mxfp8_available():
     if _is_package_available("transformer_engine", "transformer-engine"):
         import transformer_engine.pytorch as te
+
         return te.fp8.check_mxfp8_support()[0]
     return False
 
+
 def is_lomo_available():
     return _is_package_available("lomo_optim")
 

src/accelerate/utils/transformer_engine.py

@@ -146,12 +146,14 @@ def apply_fp8_autowrap(model, fp8_recipe_handler):
 
     if is_hpu_available():
         import intel_transformer_engine.recipe as te_recipe
+
         is_fp8_block_scaling_available = False
         message = "MXFP8 block scaling is not available on HPU."
 
     else:
         import transformer_engine.common.recipe as te_recipe
         import transformer_engine.pytorch as te
+
         is_fp8_block_scaling_available, message = te.fp8.check_mxfp8_support()
 
     kwargs = fp8_recipe_handler.to_kwargs() if fp8_recipe_handler is not None else {}

tests/test_fp8.py

@@ -110,7 +110,6 @@ def test_can_prepare_model_single_gpu_from_config(self):
             command += ["-m", "tests.test_fp8", "--test_te", "--from_config"]
             run_command(command)
 
-
     @require_transformer_engine_mxfp8
     def test_can_prepare_model_with_mxfp8_block_scaling(self):
         with tempfile.TemporaryDirectory() as dir_name: