infer-types

A CLI tool to automatically add type annotations into Python code.

The main scenario for using the tool is to help you with annotating a big and old codebase. It won't solve the task for you 100% but will definitely help you tremendously, because many of the functions in the real world have quite simple return types that are easy to infer automatically.

Installation

python3 -m pip install infer-types retype

This will also install retype which we're going to use to apply generated type annotations back to the code (see Usage below).

Usage

First of all, run the tool:

python3 -m infer_types ./example/

It will infer types for all code in the example directory and save stub files inside of types directory.

The next thing you need to do is to apply the stub files back to the code. For that, we're going to use retype:

retype -it ./example/ ./example/

The infer-types tool uses the new fancy syntax for type annotations introduced in Python 3.10. So, instead of Optional[str] it will emit str | None. If your code is supposed to run on an older version of Python, add from __future__ import annotations at the beginning of each file. It will solve the issue and also make startup of your app faster.

See awesome-python-typing for more tools to help you with annotating your code.