Fix UT

deepmodeling · Dec 15, 2024 · e33d6e6 · e33d6e6
1 parent 3af4970
commit e33d6e6
Show file tree

Hide file tree

Showing 5 changed files with 0 additions and 10 deletions.
diff --git a/source/tests/universal/common/cases/atomic_model/atomic_model.py b/source/tests/universal/common/cases/atomic_model/atomic_model.py
@@ -127,5 +127,3 @@ def setUpClass(cls) -> None:
         cls.aprec_dict = {}
         cls.rprec_dict = {}
         cls.epsilon_dict = {}
-        cls.property_name = ["band_prop"]
-        cls.task_dim = 3
diff --git a/source/tests/universal/common/cases/model/model.py b/source/tests/universal/common/cases/model/model.py
@@ -174,5 +174,3 @@ def setUpClass(cls) -> None:
         cls.rprec_dict = {}
         cls.epsilon_dict = {}
         cls.skip_test_autodiff = True
-        cls.property_name = ["band_prop"]
-        cls.task_dim = 3
diff --git a/source/tests/universal/dpmodel/atomc_model/test_atomic_model.py b/source/tests/universal/dpmodel/atomc_model/test_atomic_model.py
@@ -455,8 +455,6 @@ def setUpClass(cls) -> None:
             mixed_types=ds.mixed_types(),
             type_map=cls.expected_type_map,
             embedding_width=ds.get_dim_emb(),
-            task_dim=cls.task_dim,
-            property_name=cls.property_name,
         )
         ft = Fitting(
             **cls.input_dict_ft,

diff --git a/source/tests/universal/pt/atomc_model/test_atomic_model.py b/source/tests/universal/pt/atomc_model/test_atomic_model.py
@@ -445,8 +445,6 @@ def setUpClass(cls) -> None:
             mixed_types=ds.mixed_types(),
             type_map=cls.expected_type_map,
             embedding_width=ds.get_dim_emb(),
-            task_dim=cls.task_dim,
-            property_name=cls.property_name,
         )
         ft = Fitting(
             **cls.input_dict_ft,

diff --git a/source/tests/universal/pt/model/test_model.py b/source/tests/universal/pt/model/test_model.py
@@ -782,8 +782,6 @@ def setUpClass(cls) -> None:
             mixed_types=ds.mixed_types(),
             type_map=cls.expected_type_map,
             embedding_width=ds.get_dim_emb(),
-            task_dim = cls.task_dim,
-            property_name = cls.property_name,
         )
 
         ft = Fitting(