TYP: numba stub

pandas/core/_numba/kernels/shared.py

@@ -2,7 +2,13 @@
 import numpy as np
 
 
-@numba.jit(numba.boolean(numba.int64[:]), nopython=True, nogil=True, parallel=False)
+@numba.jit(
+    # error: Any? not callable
+    numba.boolean(numba.int64[:]),  # type: ignore[misc]
+    nopython=True,
+    nogil=True,
+    parallel=False,
+)
 def is_monotonic_increasing(bounds: np.ndarray) -> bool:
     """Check if int64 values are monotonically increasing."""
     n = len(bounds)

pyproject.toml

@@ -60,6 +60,7 @@ markers = [
 
 [tool.mypy]
 # Import discovery
+mypy_path = "stubs"
 namespace_packages = false
 explicit_package_bases = false
 ignore_missing_imports = true

stubs/numba.pyi

@@ -0,0 +1,41 @@
+from typing import (
+    Any,
+    Callable,
+    Literal,
+    overload,
+)
+
+import numba
+
+from pandas._typing import F
+
+def __getattr__(name: str) -> Any: ...  # incomplete
+@overload
+def jit(
+    signature_or_function: F = ...,
+) -> F: ...
+@overload
+def jit(
+    signature_or_function: str
+    | list[str]
+    | numba.core.types.abstract.Type
+    | list[numba.core.types.abstract.Type] = ...,
+    locals: dict = ...,  # TODO: Mapping of local variable names to Numba types
+    cache: bool = ...,
+    pipeline_class: numba.compiler.CompilerBase = ...,
+    boundscheck: bool | None = ...,
+    *,
+    nopython: bool = ...,
+    forceobj: bool = ...,
+    looplift: bool = ...,
+    error_model: Literal["python", "numpy"] = ...,
+    inline: Literal["never", "always"] | Callable = ...,
+    # TODO: If a callable is provided it will be called with the call expression
+    # node that is requesting inlining, the caller's IR and callee's IR as
+    # arguments, it is expected to return Truthy as to whether to inline.
+    target: Literal["cpu", "gpu", "npyufunc", "cuda"] = ...,  # deprecated
+    nogil: bool = ...,
+    parallel: bool = ...,
+) -> Callable[[F], F]: ...
+
+njit = jit