From f07c5da8e6210441d41b85b92eba0ae31f895f02 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Sun, 8 Sep 2024 17:54:32 -0400 Subject: [PATCH] fix(tests): fix `skip_dp` Signed-off-by: Jinzhe Zeng --- source/tests/consistent/descriptor/test_dpa1.py | 2 +- source/tests/consistent/descriptor/test_dpa2.py | 2 +- source/tests/consistent/descriptor/test_se_atten_v2.py | 2 +- source/tests/consistent/descriptor/test_se_t_tebd.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/source/tests/consistent/descriptor/test_dpa1.py b/source/tests/consistent/descriptor/test_dpa1.py index 955b58932a..0f44ecaae1 100644 --- a/source/tests/consistent/descriptor/test_dpa1.py +++ b/source/tests/consistent/descriptor/test_dpa1.py @@ -177,7 +177,7 @@ def skip_dp(self) -> bool: use_econf_tebd, use_tebd_bias, ) = self.param - return CommonTest.skip_pt or self.is_meaningless_zero_attention_layer_tests( + return CommonTest.skip_dp or self.is_meaningless_zero_attention_layer_tests( attn_layer, attn_dotr, normalize, diff --git a/source/tests/consistent/descriptor/test_dpa2.py b/source/tests/consistent/descriptor/test_dpa2.py index d12e1094f0..144567ae58 100644 --- a/source/tests/consistent/descriptor/test_dpa2.py +++ b/source/tests/consistent/descriptor/test_dpa2.py @@ -235,7 +235,7 @@ def skip_dp(self) -> bool: use_econf_tebd, use_tebd_bias, ) = self.param - return CommonTest.skip_pt + return CommonTest.skip_dp @property def skip_tf(self) -> bool: diff --git a/source/tests/consistent/descriptor/test_se_atten_v2.py b/source/tests/consistent/descriptor/test_se_atten_v2.py index 9ae16b96fa..989fdc16e7 100644 --- a/source/tests/consistent/descriptor/test_se_atten_v2.py +++ b/source/tests/consistent/descriptor/test_se_atten_v2.py @@ -165,7 +165,7 @@ def skip_dp(self) -> bool: use_econf_tebd, use_tebd_bias, ) = self.param - return CommonTest.skip_pt or self.is_meaningless_zero_attention_layer_tests( + return CommonTest.skip_dp or self.is_meaningless_zero_attention_layer_tests( attn_layer, attn_dotr, normalize, diff --git a/source/tests/consistent/descriptor/test_se_t_tebd.py b/source/tests/consistent/descriptor/test_se_t_tebd.py index 2ddbe70a4f..d9bd00aad3 100644 --- a/source/tests/consistent/descriptor/test_se_t_tebd.py +++ b/source/tests/consistent/descriptor/test_se_t_tebd.py @@ -116,7 +116,7 @@ def skip_dp(self) -> bool: use_econf_tebd, use_tebd_bias, ) = self.param - return CommonTest.skip_pt + return CommonTest.skip_dp @property def skip_tf(self) -> bool: