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

for more information, see https://pre-commit.ci
deepmodeling · Feb 28, 2024 · 06407b2 · 06407b2
1 parent df33191
commit 06407b2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/source/tests/pt/model/test_descriptor_se_r.py b/source/tests/pt/model/test_descriptor_se_r.py
@@ -135,9 +135,9 @@ def test_load_stat(self):
                 [
                     {
                         "r0": None,
-                        "coord": torch.from_numpy(self.coord_ext).reshape(
-                            -1, self.nall, 3
-                        ).to(env.DEVICE),
+                        "coord": torch.from_numpy(self.coord_ext)
+                        .reshape(-1, self.nall, 3)
+                        .to(env.DEVICE),
                         "atype": torch.from_numpy(self.atype_ext).to(env.DEVICE),
                         "box": None,
                         "natoms": self.nall,