Update test_se_t.py

deepmodeling · May 28, 2024 · e0d1498 · e0d1498
1 parent 7ff5eaa
commit e0d1498
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/source/tests/consistent/descriptor/test_se_t.py b/source/tests/consistent/descriptor/test_se_t.py
@@ -37,7 +37,7 @@
 
 @parameterized(
     (True, False),  # resnet_dt
-    ([],),  # excluded_types
+    ([], [[0, 1]]),  # excluded_types
     ("float32", "float64"),  # precision
     (0.0, 1e-8, 1e-2),  # env_protection
 )
@@ -90,7 +90,7 @@ def skip_tf(self) -> bool:
             precision,
             env_protection,
         ) = self.param
-        return env_protection != 0.0
+        return env_protection != 0.0 or excluded_types
 
     tf_class = DescrptSeTTF
     dp_class = DescrptSeTDP