Fix dptest UT

deepmodeling · Dec 23, 2024 · 6e428a7 · 6e428a7
1 parent 9bf75cb
commit 6e428a7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/tests/pt/test_dp_test.py b/source/tests/pt/test_dp_test.py
@@ -183,7 +183,7 @@ def test_dp_test_1_frame(self) -> None:
         pred_property = np.loadtxt(self.detail_file + ".property.out.0")[:, 1]
         np.testing.assert_almost_equal(
             pred_property,
-            to_numpy_array(result["property"])[0],
+            to_numpy_array(result[model.get_property_name()])[0],
         )
 
     def tearDown(self) -> None: