fix format

microsoft · Mar 5, 2024 · 2ea44ba · 2ea44ba
1 parent b28d81b
commit 2ea44ba
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/tests/unit/checkpoint/test_zero_optimizer.py b/tests/unit/checkpoint/test_zero_optimizer.py
@@ -56,7 +56,6 @@ def test_load_optimizer_state(self, tmpdir, zero_stage, use_cpu_offload, adam_op
         if not get_accelerator().is_fp16_supported():
             pytest.skip("fp16 is not supported")
 
-
         config_dict = {
             "train_batch_size": 2,
             "steps_per_print": 1,

diff --git a/tests/unit/compression/test_dequantization.py b/tests/unit/compression/test_dequantization.py
@@ -13,8 +13,8 @@
 from deepspeed.accelerator import get_accelerator
 
 
-
 class TestDequantization(DistributedTest):
+
     def init(self):
         local_rank = int(os.getenv("LOCAL_RANK", "0"))
         self.device = torch.device(get_accelerator().device_name(local_rank))