Type stubs for Python typing
thisiscam opened this issue · 6 comments
Hi,
I would like to generate type stubs so that we can use type checking with islpy functions.
I'm thinking of extending gen_wrap
to achieve this, but was also wondering if the maintainer has any good suggestions?
Also, it seems like the sphinx docs can already harvest the type signatures (from the generated c++ wrapper code?).
I think the most reasonable thing to do would be to implement this via pybind, cf. pybind/pybind11#945. Pybind has all the information it needs, it just needs to make it visible to the interpreter. Even better, this would allow us to just drop the documentation-focused signature gathering code in gen_wrap
.
@inducer Thanks for the input! Yes I agree that fixing it at the pybind level would be very desirable.
However, I looked into pybind a bit and the level of unfamiliarity scared me.
Since I would really like to get the type checking to work ASAP, I did a quick implementation for typestub generation in gen_wrap
: thisiscam@e14d71a.
I'm a bit stuck on how islpy
is merging the .inc
files into the final wrapper?
Currently, I have to manually concatenating the files then paste them next to islpy/__init__.py
.
That being said, I was able to get pretty good result and my IDE can finally auto-complete islpy functions.
I'm a bit stuck on how
islpy
is merging the.inc
files into the final wrapper?
They just get included, e.g. here:
islpy/src/wrapper/wrap_isl_part3.cpp
Line 5 in 33cdc99
I did a quick implementation for typestub generation in
gen_wrap
:
I could probably be talked into merging this, if you think it's of sufficient quality. (It looks like it's got some debug leftovers in it currently, such as a breakpoint()
in an exception handler.
Unfortunately, it looks like these stubs won't buy us much from the documentation angle, at least for now: sphinx-doc/sphinx#7630.
Alternatively, is it possible to set .__annotations__
on the methods from the wrapper? (I haven't tried, but cls.attr("methodname").attr("__annotations__") = ...
might work?)
I haven't tried, but cls.attr("methodname").attr("annotations") = ... might work?
I'm not really familiar with these annotation magics.
My intuition is that one would need to pass annotations as PyObject
s here (instead of annotation strings)?
pybind has py::object
that wraps and makes C++ handling of PyObject
s easier, e.g. py::make_tuple("a", "b", "c")
.