Update source/tests/consistent/descriptor/test_se_atten_v2.py

Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Jan 18, 2025 · cbf7536 · cbf7536
1 parent 473353e
commit cbf7536
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/tests/consistent/descriptor/test_se_atten_v2.py b/source/tests/consistent/descriptor/test_se_atten_v2.py
@@ -276,7 +276,7 @@ def skip_pd(self) -> bool:
             use_econf_tebd,
             use_tebd_bias,
         ) = self.param
-        return CommonTest.skip_pt or self.is_meaningless_zero_attention_layer_tests(
+        return not INSTALLED_PD or self.is_meaningless_zero_attention_layer_tests(
             attn_layer,
             attn_dotr,
             normalize,