skip case for hang

Author: Liujie0926

tests/transformers/jamba/test_modeling.py

@@ -36,7 +36,7 @@
     repeat_kv,
 )
 
-from ...testing_utils import slow
+from ...testing_utils import slow, skip_for_none_ce_case
 
 # from ..generation import GenerationTesterMixin
 from ..test_configuration_common import ConfigTester
@@ -588,7 +588,7 @@ def test_FakeMLPForwardBackward(self):
             and up_weight.grad.sum() == 0
             and down_weight.grad.sum() == 0
         )
-
+    @skip_for_none_ce_case
     def test_from_hf_hub(self):
         model_id = "ai21labs/Jamba-tiny-random"
         model = JambaForCausalLM.from_pretrained(model_id, dtype="bfloat16", from_hf_hub=True, convert_from_torch=True)

tests/utils/test_import_utils.py

@@ -15,9 +15,10 @@
 import unittest
 
 from paddlenlp.utils import install_package, uninstall_package
-
+from ..testing_utils import skip_for_none_ce_case
 
 class ImportUntilsTest(unittest.TestCase):
+    @skip_for_none_ce_case
     def test_install_specific_package(self):
         install_package("loguru", "0.6.0")
         from loguru import __version__