Fix UT

deepmodeling · Dec 15, 2024 · a1a6583 · a1a6583
1 parent 4155294
commit a1a6583
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/source/tests/consistent/fitting/test_property.py b/source/tests/consistent/fitting/test_property.py
@@ -64,7 +64,7 @@
     (0, 1),  # numb_aparam
     (True, False),  # intensive
     ("foo", ["foo"]),  # property_name
-    (3, [4], [5, 6]),  # property_dim
+    (3, [4]),  # property_dim
 )
 class TestProperty(CommonTest, FittingTest, unittest.TestCase):
     @property

diff --git a/source/tests/consistent/model/test_property.py b/source/tests/consistent/model/test_property.py
@@ -56,6 +56,8 @@ def data(self) -> dict:
             "fitting_net": {
                 "type": "property",
                 "neuron": [4, 4, 4],
+                "property_name": ["foo", "bar"],
+                "property_dim": [2,3],
                 "resnet_dt": True,
                 "numb_fparam": 0,
                 "precision": "float64",