Merge branch 'refactor_property' of github.com:Chengqian-Zhang/deepmd…

…-kit into refactor_property
deepmodeling · Dec 15, 2024 · 3249891 · 3249891
2 parents 44fa4d6 + 1cc5c37
commit 3249891
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/source/tests/pt/model/test_property_fitting.py b/source/tests/pt/model/test_property_fitting.py
@@ -74,8 +74,8 @@ def test_consistency(
                 self.nt,
                 self.dd0.dim_out,
                 task_dim=5,
-                property_name=["foo","bar"],
-                property_dim=[2,3],
+                property_name=["foo", "bar"],
+                property_dim=[2, 3],
                 numb_fparam=nfp,
                 numb_aparam=nap,
                 mixed_types=self.dd0.mixed_types(),
@@ -144,8 +144,8 @@ def test_jit(
                 self.nt,
                 self.dd0.dim_out,
                 task_dim=5,
-                property_name=["foo","bar"],
-                property_dim=[2,3],
+                property_name=["foo", "bar"],
+                property_dim=[2, 3],
                 numb_fparam=nfp,
                 numb_aparam=nap,
                 mixed_types=self.dd0.mixed_types(),
@@ -201,8 +201,8 @@ def test_trans(self) -> None:
             self.nt,
             self.dd0.dim_out,
             task_dim=11,
-            property_name=["foo","bar"],
-            property_dim=[3,8],
+            property_name=["foo", "bar"],
+            property_dim=[3, 8],
             numb_fparam=0,
             numb_aparam=0,
             mixed_types=self.dd0.mixed_types(),
@@ -268,8 +268,8 @@ def test_rot(self) -> None:
                 self.nt,
                 self.dd0.dim_out,  # dim_descrpt
                 task_dim=5,
-                property_name=["foo","bar"],
-                property_dim=[2,3],
+                property_name=["foo", "bar"],
+                property_dim=[2, 3],
                 numb_fparam=nfp,
                 numb_aparam=nap,
                 mixed_types=self.dd0.mixed_types(),
@@ -428,8 +428,8 @@ def setUp(self) -> None:
             self.nt,
             self.dd0.dim_out,
             task_dim=3,
-            property_name=["foo","bar"],
-            property_dim=[2,1],
+            property_name=["foo", "bar"],
+            property_dim=[2, 1],
             numb_fparam=0,
             numb_aparam=0,
             mixed_types=self.dd0.mixed_types(),