After Pulumi 3.113.0 we started getting Never's for pulumi applied outputs in mypy

[rshade]

What happened?

We upgraded an existing stack to pulumi 3.113.0 and mypy started throwing the following errors:

mypy __main__.py
__main__.py:15: error: Argument 1 to "all" of "Output" has incompatible type "Output[str]"; expected "Union[Never, Awaitable[Never], Output[Never]]"  [arg-type]
__main__.py:15: error: Argument 2 to "all" of "Output" has incompatible type "Output[bool]"; expected "Union[Never, Awaitable[Never], Output[Never]]"  [arg-type]
Found 2 errors in 1 file (checked 1 source file)

Example

"""A Random Python Pulumi program"""

from pulumi import Input, Output

should_print: Input[bool] = True
example_text: Input[str] = "hello World"


def example_print(text: str, should_print: bool) -> str:
    if should_print:
        print(text)
    return text


Output.all(Output.from_input(example_text), Output.from_input(should_print)).apply(
    lambda args: example_print(args[0], args[1])
)

Output of pulumi about

pulumi about
CLI
Version 3.113.3
Go Version go1.22.2
Go Compiler gc

Plugins
KIND NAME VERSION
resource aws 6.32.0
language python unknown
resource random 4.16.1

Host
OS ubuntu
Version 20.04
Arch x86_64

This project is written in python: version='3.8.10'

Dependencies:
NAME VERSION
mypy 1.9.0
pip 24.0
pkg_resources 0.0.0
pulumi_aws 6.32.0
pulumi_random 4.16.1
pyright 1.1.359
wheel 0.43.0

Additional context

No response

Contributing

Vote on this issue by adding a 👍 reaction.
To contribute a fix for this issue, leave a comment (and link to your pull request, if you've opened one already).

[Frassle]

Output.from_input seems to be the problem:

res1 = MyResource("testResource1")

reveal_type(res1.nested_numbers)
reveal_type(res1.nested_numbers.apply(lambda x: x))
reveal_type(res1.nested_numbers.apply(lambda x: x["foo"]))
reveal_type(Output.from_input(res1.nested_numbers))
reveal_type(Output.from_input(res1.nested_numbers).apply(lambda x: x))

lib/test/langhost/output_nested/__main__.py:48: note: Revealed type is "pulumi.output.Output[builtins.dict[Any, Any]]"
lib/test/langhost/output_nested/__main__.py:49: note: Revealed type is "pulumi.output.Output[builtins.dict[Any, Any]]"
lib/test/langhost/output_nested/__main__.py:50: note: Revealed type is "pulumi.output.Output[Any]"
lib/test/langhost/output_nested/__main__.py:51: note: Revealed type is "pulumi.output.Output[Never]"
lib/test/langhost/output_nested/__main__.py:52: note: Revealed type is "pulumi.output.Output[Never]"