chore: refactor global stat

deepmodeling · Apr 8, 2024 · 80159e1 · 80159e1
1 parent 6c299fa
commit 80159e1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/source/tests/pt/model/test_linear_atomic_model_stat.py b/source/tests/pt/model/test_linear_atomic_model_stat.py
@@ -154,6 +154,7 @@ def setUp(self):
                 ),
                 # bias of foo: 1, 3
                 "energy": to_torch_tensor(np.array([5.0, 7.0]).reshape(2, 1)),
+                "find_energy": np.float32(1.0),
             }
         ]
         self.tempdir = tempfile.TemporaryDirectory()