[stubgen] Union is not defined

[matejsp]

Another stubgen issue

Stubgen for following code:

def get_country_code(country_code):
    # type: (str) -> str | None

generated:

def get_country_code(country_code: str) -> Union[str, None]: ...

But Union is not defined as import.

Your Environment

• Mypy version used: v0.981
• Mypy command-line flags: /
• Mypy configuration options from mypy.ini (and other config files): /
• Python version used: 3.8.13

[hauntsaninja]

Duplicate of #12929