✨ special ufunc type signature
#27
Assignees
Labels
Comments
Merged
|
The
Note that The number of np.single | np.double | np.longdouble | np.csingle | np.cdoubleWhich is (almost always) equivalent to np.float32 | np.float64 | np.longdouble | np.complex64 | np.complex128The dtype of the input must have the following scalar type: np.long | np.single | np.double | np.longdouble | np.csingle | np.cdouble(on Which is (almost always) equivalent to np.long | np.float32 | np.float64 | np.longdouble | np.complex64 | np.complex128
These are the 233
|
47 tasks
|
ufunc counts per signature |
jorenham
added
status: in progress
and removed
meta: codegen
jorenham
changed the title
ufuncs in scipy.specialscipy.special
jorenham
changed the title
scipy.specialscipy.special ufunc annotations
jorenham
changed the title
scipy.special ufunc annotationsscipy.special ufunc annotations (80% done)
jorenham
changed the title
scipy.special ufunc annotations (80% done)special ufunc annotations (80% done)
This was referenced Dec 23, 2024
jorenham
changed the title
special ufunc annotations (80% done)special ufunc annotations
jorenham
changed the title
special ufunc annotationsspecial ufunc type signature
|
damn, that sucked... let's just hope it won't require a lot of maintenance of bug-fixing 😅 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
The majority of the
scipy.specialfunctions arenumpy.ufunc(230 / 356 ~= 64.6%).But the
numpy.ufuncisn't generic, making most of the type signatures inscipy.specialmaximally loose, i.e.(*args, **kwds) -> object.In NumPy's bundled stubs, there's a workaround that uses typ-check-only
ufunc"subtypes" for each of the possibleninandnargscombinations (at mostnin=2andnout=2).We apply a similar approach, but more specific to each of the type signatures.
ufunc properties
__name__ninnoutnargsntypestypesidentitysignatureufunc methods
__call__atreduceaccumulatereduceatouterufunc types (signatures)
{'f->f', 'd->d'}{'f->f', 'd->d', 'g->g'}{'f->f', 'd->d', 'F->F', 'D->D'}{'F->F', 'D->D'}{'ld->d'}{'ff->f', '(l|d)d->d'}{'fF->F', 'dD->D'}{'ff->f', 'dd->d', 'fF->F', 'dD->D'}{'ff->f', 'dd->d', 'FF->F', 'DD->D'}{'fff->f', '(l|d)(l|d)d->d'}{'fff->f', '(l|d)dd->d', 'ffF->F', 'ddD->D'}{'fff->f', 'ddd->d', 'FFF->F', 'DDD->D'}{'ffff->f', 'dddd->d'}{'(qq|ff)ff->F', '(qq)dd->D'}(sph_harm){'ffff->f', '(l|d)ddd->d', 'fffF->F', 'dddD->D'}{'ffff->f', 'dddd->d', 'FFFF->F', 'DDDD->D'}{'f->ff', 'd->dd'}{'f->FF', 'd->DD'}{'f->ff', 'd->dd', 'F->FF', 'D->DD'}{'ff->ff', 'dd->dd'}{'fff->ff'. 'ddd->dd'}{'ffff->ff', 'dddd->dd'}{'fffff->ff', 'ddddd->dd'}{'f->ffff', 'd->dddd'}{'f->FFFF', 'd->DDDD'}{'f->ffff', 'd->dddd', 'F->FFFF', 'D->DDDD'}{'ff->ffff', 'dd->dddd'}The text was updated successfully, but these errors were encountered: