diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_bert.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_bert.py index 6c26189a4adb37579bb2916a82a00df81f96a125..a9e94ef09b9ac0485b8c13d73a43558fd9edf163 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_bert.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_bert.py @@ -220,4 +220,5 @@ class TestBert(unittest.TestCase): if __name__ == '__main__': - unittest.main() + with fluid.framework._test_eager_guard(): + unittest.main() diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_bmn.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_bmn.py index f69abb1e376694dddedf0259ecaff7e33c1e7838..00af9c96ba9cc4405b3260a97cfcadbd4026e6bf 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_bmn.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_bmn.py @@ -773,4 +773,5 @@ class TestTrain(unittest.TestCase): if __name__ == "__main__": - unittest.main() + with fluid.framework._test_eager_guard(): + unittest.main() diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_mobile_net.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_mobile_net.py index c6f491a5484d9f1601993e1dd581e46008f0e27a..f58041cbb6c8d629ff56749af9c2e69d43edf4c0 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_mobile_net.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_mobile_net.py @@ -628,4 +628,5 @@ class TestMobileNet(unittest.TestCase): if __name__ == '__main__': - unittest.main() + with fluid.framework._test_eager_guard(): + unittest.main() diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_program_translator.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_program_translator.py index d2c43c31a8839a4acfc66feb981b5e4309d66e42..b0ffbac88fb42921a744d046665f56b1221537e1 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_program_translator.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_program_translator.py @@ -363,4 +363,5 @@ class TestRemoveCommentInDy2St(unittest.TestCase): if __name__ == '__main__': - unittest.main() + with fluid.framework._test_eager_guard(): + unittest.main() diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_sentiment.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_sentiment.py index 74d415cc3eed7f96c15c755177ef798fd43a3a1b..b72894fb14764e2aed79d88db506a82bc2bec059 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_sentiment.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_sentiment.py @@ -358,4 +358,5 @@ class TestSentiment(unittest.TestCase): if __name__ == '__main__': - unittest.main() + with fluid.framework._test_eager_guard(): + unittest.main() diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_word2vec.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_word2vec.py index dd6cc7e9d32063ec4f7d8975028262b33a37640d..f270c5672afc3484f0ce8fcf9b49f2e0d3ab2acf 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_word2vec.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_word2vec.py @@ -303,4 +303,5 @@ class TestWord2Vec(unittest.TestCase): if __name__ == '__main__': - unittest.main() + with fluid.framework._test_eager_guard(): + unittest.main() diff --git a/python/paddle/fluid/tests/unittests/test_jit_save_load.py b/python/paddle/fluid/tests/unittests/test_jit_save_load.py index fc58f979b4dc2de42fd58a0c39ed9a16d30aee99..5dabf854734dd5e5e82d67f59ea0be3a607ef55c 100644 --- a/python/paddle/fluid/tests/unittests/test_jit_save_load.py +++ b/python/paddle/fluid/tests/unittests/test_jit_save_load.py @@ -1454,4 +1454,5 @@ class TestInputSpecCompatibility(unittest.TestCase): if __name__ == '__main__': - unittest.main() + with fluid.framework._test_eager_guard(): + unittest.main()