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

for more information, see https://pre-commit.ci
deepmodeling · Apr 10, 2024 · 2aff780 · 2aff780
1 parent 774f2ce
commit 2aff780
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/deepmd/common.py b/deepmd/common.py
@@ -126,7 +126,7 @@ def add_data_requirement(
     # standardize key
     if "atomic_" in key:
         key = key.replace("atomic_", "atom_")
-        
+
     data_requirement[key] = {
         "ndof": ndof,
         "atomic": atomic,

diff --git a/deepmd/utils/data.py b/deepmd/utils/data.py
@@ -180,7 +180,7 @@ def add(
         # standardize key
         if "atomic_" in key:
             key = key.replace("atomic_", "atom_")
-            
+
         self.data_dict[key] = {
             "ndof": ndof,
             "atomic": atomic,

diff --git a/source/tests/pt/model/test_polar_stat.py b/source/tests/pt/model/test_polar_stat.py
@@ -62,9 +62,9 @@ def test_atomic_consistency(self):
 
     def test_global_consistency(self):
         self.sampled[0]["find_atom_polarizability"] = -1
-        self.sampled[0]["polarizability"] = self.sampled[0][
-            "atom_polarizability"
-        ].sum(dim=1)
+        self.sampled[0]["polarizability"] = self.sampled[0]["atom_polarizability"].sum(
+            dim=1
+        )
         self.all_stat["find_atom_polarizability"] = [-1]
         self.all_stat["polarizability"] = [
             self.all_stat["atom_polarizability"][0].sum(axis=1)