Apply formatting (#929)

* Apply formatting
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Apply formatting
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

---------
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

tests/paddle/parallel_tests/amax_reduction.py

@@ -27,7 +27,7 @@ class TestAmaxReduction(unittest.TestCase):
     def setUp(self):
         self.data_parallel_size = 2
         self.init_dist_env()
-        self.global_dtype = 'bfloat16'
+        self.global_dtype = "bfloat16"
         paddle.set_default_dtype(self.global_dtype)
 
     def init_dist_env(self):
@@ -83,5 +83,5 @@ class TestAmaxReduction(unittest.TestCase):
             assert_allclose_across_ranks(layer2.fp8_meta["scaling_bwd"].scale_inv)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     unittest.main()

tests/paddle/test_sanity_import.py

@@ -3,4 +3,5 @@
 # See LICENSE for license information.
 
 import transformer_engine.paddle
+
 print("OK")