[Dy2Stat]Filter UserWarings while @to_static (PaddlePaddle#45754)

* [Dy2Stat]Filter UserWarings while @to_static * only filter DeprecationWarning * fix unittest
Caozhou1995 · Sep 8, 2022 · 5f9e0b3 · 5f9e0b3
1 parent 2190212
commit 5f9e0b3
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/python/paddle/fluid/dygraph/dygraph_to_static/utils.py b/python/paddle/fluid/dygraph/dygraph_to_static/utils.py
@@ -609,7 +609,8 @@ def _inject_import_statements():
     import_statements = [
         "import paddle", "from paddle import Tensor",
         "import paddle.fluid as fluid", "import paddle.jit.dy2static as _jst",
-        "from typing import *", "import numpy as np"
+        "from typing import *", "import numpy as np", "import warnings",
+        "warnings.filterwarnings('ignore', category=DeprecationWarning)"
     ]
     return '\n'.join(import_statements) + '\n'
 

diff --git a/python/paddle/fluid/layers/math_op_patch.py b/python/paddle/fluid/layers/math_op_patch.py
@@ -378,7 +378,8 @@ def __impl__(self, other_var):
                     "If your code works well in the older versions but crashes in this version, try to use "
                     "%s(X, Y, axis=0) instead of %s. This transitional warning will be dropped in the future."
                     % (file_name, line_num, EXPRESSION_MAP[method_name],
-                       op_type, op_type, EXPRESSION_MAP[method_name]))
+                       op_type, op_type, EXPRESSION_MAP[method_name]),
+                    category=DeprecationWarning)
             current_block(self).append_op(type=op_type,
                                           inputs={
                                               'X': [self],

diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_origin_info.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_origin_info.py
@@ -65,7 +65,7 @@ def set_test_func(self):
         self.func = simple_func
 
     def set_static_lineno(self):
-        self.static_abs_lineno_list = [7, 8, 9]
+        self.static_abs_lineno_list = [9, 10, 11]
 
     def set_dygraph_info(self):
         self.line_num = 3
@@ -149,7 +149,7 @@ def set_test_func(self):
         self.func = nested_func
 
     def set_static_lineno(self):
-        self.static_abs_lineno_list = [7, 9, 10, 11, 12]
+        self.static_abs_lineno_list = [9, 11, 12, 13, 14]
 
     def set_dygraph_info(self):
         self.line_num = 5
@@ -174,7 +174,7 @@ def set_test_func(self):
         self.func = decorated_func
 
     def set_static_lineno(self):
-        self.static_abs_lineno_list = [7, 8]
+        self.static_abs_lineno_list = [9, 10]
 
     def set_dygraph_info(self):
         self.line_num = 2
@@ -208,7 +208,7 @@ def set_test_func(self):
         self.func = decorated_func2
 
     def set_static_lineno(self):
-        self.static_abs_lineno_list = [7, 8]
+        self.static_abs_lineno_list = [9, 10]
 
     def set_dygraph_info(self):
         self.line_num = 2