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

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

deepmd/pt/utils/dataloader.py

@@ -62,14 +62,15 @@ class DpLoaderSet(Dataset):
     sys_path
             Path to the data system
     batch_size
-            Max frame count in a batch. 
+            Max frame count in a batch.
     type_map
             Gives the name of different atom types
-    seed    
+    seed
             Random seed for dataloader
     shuffle
             If the data are shuffled (Only effective in serial mode. Always shuffle in distributed data parallelism)
     """
+
     def __init__(
         self,
         systems,

deepmd/pt/utils/dataset.py

@@ -16,11 +16,7 @@
 
 
 class DeepmdDataSetForLoader(Dataset):
-    def __init__(
-        self,
-        system: str,
-        type_map: str
-    ):
+    def __init__(self, system: str, type_map: str):
         """Construct DeePMD-style dataset containing frames cross different systems.
 
         Args:
@@ -29,9 +25,7 @@ def __init__(
         """
         self.system = system
         self._type_map = type_map
-        self._data_system = DeepmdData(
-            sys_path=system,type_map=self._type_map
-        )
+        self._data_system = DeepmdData(sys_path=system, type_map=self._type_map)
         self.mixed_type = self._data_system.mixed_type
         self._ntypes = self._data_system.get_ntypes()
         self._natoms = self._data_system.get_natoms()