make style

byi8220 · Jul 15, 2024 · 6bfe871 · 6bfe871
1 parent 8d2c6c3
commit 6bfe871
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/src/accelerate/data_loader.py b/src/accelerate/data_loader.py
@@ -21,7 +21,6 @@
 
 from accelerate.utils import is_torchdata_stateful_dataloader_available
 
-
 from .logging import get_logger
 from .state import AcceleratorState, DistributedType, GradientState, is_torch_xla_available
 from .utils import (

diff --git a/tests/test_accelerator.py b/tests/test_accelerator.py
@@ -35,7 +35,7 @@
     require_non_torch_xla,
     require_torchdata_stateful_dataloader,
 )
-from accelerate.utils import patch_environment, is_torchdata_stateful_dataloader_available
+from accelerate.utils import is_torchdata_stateful_dataloader_available, patch_environment
 from accelerate.utils.dataclasses import DataLoaderConfiguration
 from accelerate.utils.modeling import get_state_dict_from_offload, load_checkpoint_in_model
 from accelerate.utils.random import set_seed

diff --git a/tests/test_data_loader.py b/tests/test_data_loader.py
@@ -31,8 +31,8 @@
     skip_first_batches,
 )
 from accelerate.test_utils.testing import require_torchdata_stateful_dataloader
-from accelerate.utils.dataclasses import DataLoaderConfiguration
 from accelerate.utils import is_torchdata_stateful_dataloader_available
+from accelerate.utils.dataclasses import DataLoaderConfiguration
 
 
 if is_torchdata_stateful_dataloader_available():