From e0d14988800aa54d7513ae606db932a43aa88827 Mon Sep 17 00:00:00 2001
From: Duo <50307526+iProzd@users.noreply.github.com>
Date: Tue, 28 May 2024 21:33:52 +0800
Subject: [PATCH] Update test_se_t.py

---
 source/tests/consistent/descriptor/test_se_t.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/source/tests/consistent/descriptor/test_se_t.py b/source/tests/consistent/descriptor/test_se_t.py
index 5b90077f17..7579344012 100644
--- 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