CI: Include all modules into pytype checking

PiperOrigin-RevId: 688522741
google · Oct 28, 2024 · 6df9d95 · 6df9d95
1 parent 9852e85
commit 6df9d95
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 12 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,19 +4,14 @@ build-backend = "setuptools.build_meta"
 
 [tool.pytype]
 inputs = [
-    "pytype/*.py",
-    "pytype/overlays/",
-    "pytype/pyc/",
-    "pytype/pyi/",
-    "pytype/pytd/",
-    "pytype/tools/",
-    "pytype/typegraph/",
+    "pytype/**/*.py",
     "pytype_extensions/**/*.py",
 ]
 exclude = [
     "**/*_test.py",
     "**/test_*.py",
     "**/*_test_*.py",
-    "pytype/tools/merge_pyi/test_data/",
-    "pytype/tools/xref/testdata/",
-]
+    "**/typeshed/",
+    "**/test_data/",
+    "**/testdata/",
+]
diff --git a/pytype/abstract/_classes.py b/pytype/abstract/_classes.py
@@ -692,7 +692,9 @@ class FunctionPyTDClass(PyTDClass):
   """
 
   def __init__(
-      self, func: "_function_base.Function", ctx: "context.Context"
+      self,
+      func: "_function_base.Function | _function_base.BoundFunction",
+      ctx: "context.Context",
   ) -> None:
     super().__init__(
         "typing.Callable",

diff --git a/pytype/imports/pickle_utils.py b/pytype/imports/pickle_utils.py
@@ -103,7 +103,9 @@ def DecodeAst(data: bytes) -> serialize_ast.SerializableAst:
 def LoadAst(
     filename: Path, compress: bool = False, open_function=open
 ) -> serialize_ast.SerializableAst:
-  return _Load(AstDecoder, filename, compress, open_function)
+  return _Load(
+      AstDecoder, filename, compress, open_function
+  )  # pytype: disable=bad-return-type
 
 
 def DecodeBuiltins(data: bytes) -> serialize_ast.ModuleBundle: