Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Stubgen: guess return types based on returned values #18116

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

yangdanny97
Copy link
Contributor

This PR extends stubgen to guess return types for some functions based on the returned expression. This is inspired by autotyping's scalar-return flag, with the addition of being able to handle literal lists, tuples, sets, and dicts.

There are some limitations:

This only works on concrete functions at least one return statement & no yield/yieldfroms. It only works on literals, tuples of literals, or homogeneous container literals. Additionally, it only works if all the return statements return the same type or None.

To verify this behavior, I added a test case with some examples and updated all the existing stubtest cases.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant