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

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

Author: pre-commit-ci[bot]

deepmd/entrypoints/test.py

@@ -1138,14 +1138,13 @@ def test_polar(
     )
     # for TF compatibility, need to add type_sel to coord, so that the atom order is aligned with the label
     data.add(
-        "coord", 
-        3, 
-        atomic=True, 
+        "coord",
+        3,
+        atomic=True,
         must=True,
         type_sel=dp.get_sel_type(),
     )
 
-
     test_data = data.get_test()
     polar, numb_test, atype = run_test(dp, test_data, numb_test, data)
 
@@ -1288,9 +1287,9 @@ def test_dipole(
     )
     # for TF compatibility, need to add type_sel to coord, so that the atom order is aligned with the label
     data.add(
-        "coord", 
-        3, 
-        atomic=True, 
+        "coord",
+        3,
+        atomic=True,
         must=True,
         type_sel=dp.get_sel_type(),
     )

deepmd/utils/data.py

@@ -637,7 +637,9 @@ def _load_data(
             data = path.load_numpy().astype(dtype)
             try:  # YWolfeee: deal with data shape error
                 if atomic:
-                    if type_sel is not None and natoms_sel != natoms: # handle PT all atom case
+                    if (
+                        type_sel is not None and natoms_sel != natoms
+                    ):  # handle PT all atom case
                         # check the data shape is nsel or natoms
                         if data.size == nframes * natoms_sel * ndof_:
                             if output_natoms_for_type_sel: