update format

tests/test_save_checkpoint.py

@@ -1,14 +1,15 @@
 import os
 import shutil
-import pytest
 
+import pytest
 import torch.distributed as dist
 from torch.distributed.fsdp import FullyShardedDataParallel as FSDP
 
-from fastvideo.utils.checkpoint import save_checkpoint
 from fastvideo.models.mochi_hf.modeling_mochi import MochiTransformer3DModel
+from fastvideo.utils.checkpoint import save_checkpoint
 from fastvideo.utils.fsdp_util import get_dit_fsdp_kwargs
 
+
 @pytest.fixture(scope="module", autouse=True)
 def setup_distributed():
     os.environ["RANK"] = "0"
@@ -21,6 +22,7 @@ def setup_distributed():
     yield
     dist.destroy_process_group()
 
+
 def test_save_and_remove_checkpoint():
     transformer = MochiTransformer3DModel(num_layers=0)
     fsdp_kwargs, _ = get_dit_fsdp_kwargs(transformer, "none")