[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

deepmd/pd/train/training.py

@@ -23,9 +23,6 @@
 from paddle.framework import (
     core,
 )
-from paddle.io import (
-    DataLoader,
-)
 
 from deepmd.common import (
     symlink_prefix_files,
@@ -58,7 +55,6 @@
     DEFAULT_PRECISION,
     DEVICE,
     JIT,
-    NUM_WORKERS,
     SAMPLER_RECORD,
     enable_prim,
 )
@@ -180,8 +176,12 @@ def get_dataloader_and_buffer(_data, _params):
                 # )
                 # _data_buffered = BufferedIterator(iter(_dataloader))
                 # return _dataloader, _data_buffered
-                
-                from itertools import chain, cycle
+
+                from itertools import (
+                    chain,
+                    cycle,
+                )
+
                 all_dataloaders = []
                 self.all_dlen = 0
                 for dataloader in _data.dataloaders:
@@ -192,7 +192,9 @@ def get_dataloader_and_buffer(_data, _params):
                     self.all_dlen += dlen
                     all_dataloaders.append(shard_dataloader)
                 _shard_dataloader = cycle(chain(*all_dataloaders))
-                _data_buffered = BufferedIterator(iter(_shard_dataloader),self.all_dlen)
+                _data_buffered = BufferedIterator(
+                    iter(_shard_dataloader), self.all_dlen
+                )
                 return _shard_dataloader, _data_buffered
 
             training_dataloader, training_data_buffered = get_dataloader_and_buffer(

deepmd/pd/utils/dataloader.py

@@ -182,8 +182,7 @@ def construct_dataset(system):
             system_batch_sampler = BatchSampler(
                 system,
                 shuffle=(
-                    (not (dist.is_available() and dist.is_initialized()))
-                    and shuffle
+                    (not (dist.is_available() and dist.is_initialized())) and shuffle
                 ),
                 batch_size=int(batch_size),
             )