From e33d6e68f89e37b70da169d6769737fe7d3eb9a2 Mon Sep 17 00:00:00 2001
From: root <2000011006@stu.pku.edu.cn>
Date: Sun, 15 Dec 2024 19:30:06 +0800
Subject: [PATCH] Fix UT

---
 .../tests/universal/common/cases/atomic_model/atomic_model.py   | 2 --
 source/tests/universal/common/cases/model/model.py              | 2 --
 source/tests/universal/dpmodel/atomc_model/test_atomic_model.py | 2 --
 source/tests/universal/pt/atomc_model/test_atomic_model.py      | 2 --
 source/tests/universal/pt/model/test_model.py                   | 2 --
 5 files changed, 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
index 2c0c27d7ad..499a313a32 100644
--- 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
index 423f3bc538..cee69d9d6c 100644
--- 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
index 5a61a8d554..7b579ae82c 100644
--- 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
index 40e846f81a..f41d384b6b 100644
--- 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
index 0a38d5e9df..1663412ffd 100644
--- 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(