Omit Return Types for self in Python Output

[rirze]

❓ Guidance

Affected Languages

• TypeScript or Javascript
• Python
• Java
• .NET (C#, F#, ...)
• Go

The Question

I'm using a library that is written in Typescript but crosspiles to Python using jsii. While the Python API works, I would like the language server to behave nicely with the written Python code. In this specific instance, the problem can boiled down to this example that mimics the Python output code from jsii:

class Base:
    def foo(self) -> "Base":
        return self

class SubBase(Base):
    def bar(self) -> "SubBase":
        return self

Then if I type the line:

The language server complains, because the return type of Base.foo is Base and so it doesn't recognize .bar as a valid method for the object.

Alternatively, if I write these class definitions without the explicit return type for self:

class Base:
    def foo(self):
        return self

class SubBase(Base):
    def bar(self):
        return self

I get no red lines from the language server, in addition to getting the proper autocomplete options which show both classes' methods.

To reiterate, the code in the image works when I execute the file, I am looking to getting the resulting Python code to work nicely with the language server and show results intuitively to anyone using the library.

Is there a way for jsii to omit the return types for this scenario, for where the original TS code does not specify a return type as well?