jorenham

Rank-typing coverage of the public NumPy API.

numpy

Two things that cut across many rows. The nan* functions are stub aliases of their non-nan counterparts and are fixed together. For the reductions, axis=None and keepdims=True already give a known rank and axis=int is the gap, except where marked otherwise.

Function	Progress	Effort	Limit	Uses (x1000)
np.array[^nested]	◐	M		20,218
np.arange	●			10,060

jorenham

1 -> 1

ufunc	identity	arg	out
isnat		Mm	?
signbit		efdg	?
isfinite		?bBhHiIlLqQefdgFDGmM	?
isinf		?bBhHiIlLqQefdgFDGmM	?
isnan		?bBhHiIlLqQefdgFDGmMT	?
bitwise_count		bBhHiIlLqQO	BO

jorenham

note: unused allowlist entry scipy\._lib\.(array_api_(compat|extra)|cobyqa|pyprima).*
note: unused allowlist entry scipy\.fft\._pocketfft\..*
scipy.__all__
scipy._lib._array_api.__all__
scipy._lib._array_api.array_namespace
scipy._lib._array_api.is_pydata_sparse_array
scipy._lib._array_api.make_xp_pytest_param
scipy._lib._array_api.xp_capabilities
scipy._lib._array_api_compat_vendor
scipy._lib._array_api_docs_tables.is_inherently_out_of_scope

jorenham

from typing import Protocol, Literal, reveal_type


class CanRMul[InT, OutT](Protocol):
    def __rmul__(self, other: InT, /) -> OutT: ...


def twice[OutT](x: CanRMul[Literal[2], OutT]) -> OutT:
    return 2 * x

jorenham

{
  "label": "",
  "message": "ty",
  "logoSvg": "<svg xmlns=\"http://www.w3.org/2000/svg\" width=\"48\" height=\"48\"><path d=\"M48 7.7H27.8V0h-24v7.7H0v18.2h3.8v14.3c0 4.3 3.5 7.8 7.8 7.8H48V29.8H27.8v-3.9h12.4c4.3 0 7.8-3.5 7.8-7.8V7.7Z\" fill=\"#46ebe1\"/></svg>",
  "logoWidth": 10,
  "labelColor": "grey",
  "color": "#261230"
}

jorenham

jorenham

trait Family {
    type Member<T>;
}

trait Functor<A> {
    type Of: Family<Member<A> = Self>;
    fn fmap<B>(self, f: impl FnMut(A) -> B) -> <Self::Of as Family>::Member<B>;
}

//

jorenham

import subprocess
from pathlib import Path

# see https://github.com/astral-sh/ruff/pull/18224
from ruff.__main__ import (  # type: ignore[import-untyped]  # pyright: ignore[reportMissingTypeStubs]
    find_ruff_bin,  # noqa: PLC2701
)


def ruff_format(source: str) -> str:

jorenham

original post: scipy/scipy-stubs#153 (comment)

---

Annotating a function whose return type depends on an optional positional- or keyword-only parameter in Python

Overloads are roughly similar to pattern matching[^1], but from the perspective of the caller.

Consider this function for example

jorenham

Without scipy-stubs	With scipy-stubs

• VSCode 1.96.2
• PyLance 2024.12.100
• pyright == 1.1.391
• scipy == 1.15.0