[Typing] 识别标题中的 debug 关键字进行类型检查 (#65319)

megemini · web-flow · commit 6498a8920883 · 2024-06-21T10:42:50.000+08:00
diff --git a/paddle/scripts/paddle_build.sh b/paddle/scripts/paddle_build.sh
@@ -3562,11 +3562,20 @@ function exec_type_checking() {
 
     # check all sample code
     TITLE_CHECK_ALL=`curl -s https://github.com/PaddlePaddle/Paddle/pull/${GIT_PR_ID} | grep "<title>" | grep -i "typing all" || true`
+    DEBUG_MODE=`curl -s https://github.com/PaddlePaddle/Paddle/pull/${GIT_PR_ID} | grep "<title>" | grep -i "[debug]" || true`
 
     if [[ ${TITLE_CHECK_ALL} ]]; then
-        python type_checking.py --full-test; type_checking_error=$?
+        if [[ ${DEBUG_MODE} ]]; then
+            python type_checking.py --debug --full-test; type_checking_error=$?
+        else
+            python type_checking.py --full-test; type_checking_error=$?
+        fi
     else
-        python type_checking.py; type_checking_error=$?
+        if [[ ${DEBUG_MODE} ]]; then
+            python type_checking.py --debug; type_checking_error=$?
+        else
+            python type_checking.py; type_checking_error=$?
+        fi
     fi
 
     if [ "$type_checking_error" != "0" ];then
diff --git a/python/paddle/_typing/__init__.py b/python/paddle/_typing/__init__.py
@@ -23,6 +23,7 @@
     NestedStructure as NestedStructure,
     Numberic as Numberic,
     NumbericSequence as NumbericSequence,
+    PaddingMode as PaddingMode,
     TensorIndex as TensorIndex,
     TensorLike as TensorLike,
     TensorOrTensors as TensorOrTensors,
diff --git a/python/paddle/nn/functional/pooling.py b/python/paddle/nn/functional/pooling.py
@@ -12,6 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
 import numpy as np
 
 from paddle import _C_ops, in_dynamic_mode
@@ -33,6 +37,11 @@
     convert_to_list,
 )
 
+if TYPE_CHECKING:
+    from paddle import Tensor
+
+    from ..._typing import PaddingMode, Size1, Size2
+
 __all__ = []
 
 
@@ -173,14 +182,14 @@ def _expand_low_nd_padding(padding):
 
 
 def avg_pool1d(
-    x,
-    kernel_size,
-    stride=None,
-    padding=0,
-    exclusive=True,
-    ceil_mode=False,
-    name=None,
-):
+    x: Tensor,
+    kernel_size: Size1,
+    stride: Size1 | None = None,
+    padding: PaddingMode | Size1 | Size2 = 0,
+    exclusive: bool = True,
+    ceil_mode: bool = False,
+    name: str | None = None,
+) -> Tensor:
     """
     This API implements average pooling 1d operation,
     See more details in :ref:`api_paddle_nn_AvgPool1d` .
diff --git a/tools/type_checking.py b/tools/type_checking.py
@@ -70,10 +70,16 @@ class TestResult:
 
 class MypyChecker(TypeChecker):
     def __init__(
-        self, config_file: str, cache_dir: str, *args: Any, **kwargs: Any
+        self,
+        config_file: str,
+        cache_dir: str,
+        debug: bool = False,
+        *args: Any,
+        **kwargs: Any,
     ) -> None:
         self.config_file = config_file
         self.cache_dir = cache_dir
+        self.debug = debug
         super().__init__(*args, **kwargs)
 
     def run(self, api_name: str, codeblock: str) -> TestResult:
@@ -99,7 +105,8 @@ def run(self, api_name: str, codeblock: str) -> TestResult:
         )
 
         normal_report, error_report, exit_status = mypy_api.run(
-            [
+            (["--show-traceback"] if self.debug else [])
+            + [
                 f'--config-file={self.config_file}',
                 f'--cache-dir={self.cache_dir}',
                 '-c',
@@ -313,5 +320,6 @@ def run_type_checker(
         cache_dir=(
             args.cache_dir if args.cache_dir else (base_path / '.mypy_cache')
         ),
+        debug=args.debug,
     )
     run_type_checker(args, mypy_checker)
