only filter DeprecationWarning

PaddlePaddle · Sep 5, 2022 · 3ddaaa1 · 3ddaaa1
1 parent 154dc75
commit 3ddaaa1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/python/paddle/fluid/dygraph/dygraph_to_static/utils.py b/python/paddle/fluid/dygraph/dygraph_to_static/utils.py
@@ -610,7 +610,7 @@ def _inject_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", "import warnings",
-        "warnings.filterwarnings('ignore')"
+        "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],