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

for more information, see https://pre-commit.ci
deepmodeling · Jan 28, 2024 · 8fee8fa · 8fee8fa
1 parent 84767f3
commit 8fee8fa
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/deepmd/pt/model/model/pair_tab.py b/deepmd/pt/model/model/pair_tab.py
@@ -222,10 +222,7 @@ def _check_table_upper_boundary(self):
                     upper, increment * (rcut_idx + 1), rcut_idx - upper_idx + 1
                 )
                 pad_linear[:-1, 1:] = np.array(
-                    [
-                        np.linspace(start, 0, rcut_idx - upper_idx)
-                        for start in upper_val
-                    ]
+                    [np.linspace(start, 0, rcut_idx - upper_idx) for start in upper_val]
                 ).T
                 raw_data = np.concatenate((raw_data[:-1, :], pad_linear), axis=0)
 

diff --git a/source/tests/pt/test_pairtab.py b/source/tests/pt/test_pairtab.py
@@ -56,7 +56,7 @@ def test_without_mask(self):
         )
         expected_result = torch.tensor([[1.2000, 1.3542], [1.2000, 0.4000]])
 
-        torch.testing.assert_allclose(result['energy'], expected_result)
+        torch.testing.assert_allclose(result["energy"], expected_result)
 
     def test_with_mask(self):
         self.nlist = torch.tensor([[[1, -1], [0, 2]], [[1, 2], [0, 3]]])
@@ -66,7 +66,7 @@ def test_with_mask(self):
         )
         expected_result = torch.tensor([[0.8000, 1.3542], [1.2000, 0.4000]])
 
-        torch.testing.assert_allclose(result['energy'], expected_result)
+        torch.testing.assert_allclose(result["energy"], expected_result)
 
     def test_jit(self):
         model = torch.jit.script(self.model)