Merge pull request PaddlePaddle#9 from 0x45f/fix-ut1

test warning ast only
0x45f · Jun 19, 2023 · c2fe354 · c2fe354
2 parents 9d5c54f + fdc47ac
commit c2fe354
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test/dygraph_to_static/test_warning.py b/test/dygraph_to_static/test_warning.py
@@ -15,6 +15,8 @@
 import unittest
 import warnings
 
+from dygraph_to_static_util import ast_only_test, dy2static_unittest
+
 import paddle
 from paddle.static.nn import cond
 
@@ -37,12 +39,14 @@ def false_fn():
     return [paddle.to_tensor(3), [None, paddle.to_tensor(4)]]
 
 
+@dy2static_unittest
 class TestReturnNoneInIfelse(unittest.TestCase):
+    @ast_only_test
     def test_dy2static_warning(self):
         paddle.disable_static()
         with warnings.catch_warnings(record=True) as w:
             warnings.simplefilter("always")
-            fun1()
+            paddle.jit.to_static(fun1)()
             flag = False
             for warn in w:
                 if (