Environment yaml arg type hints don't indicate all accepted inputs

[stooj]

What happened?

The docs and type hints state that the Environment resource expects pulumi.Asset which cannot handle outputs.

This means you can't have outputs in the yaml.

If you ignore the type hints, you can input an output (see example below) and it works fine but breaks type checking. Could you please check and update the type hints?

Example

def callback(arg):
    data = {
        'values': {
            'root': {
                'cloud-config': {
                    'foo': arg,
                },
            },
        },
    }
    return yaml.safe_dump(data)

# aws_cloud.testexport is a random aws resource

Environment(
    'public',
    organization=pulumi.get_organization(),
    project=pulumi.get_project(),
    name='public',
    yaml=aws_cloud.testexport.arn.apply(callback),
)

Additional context

Here's a version that works and gets type checked successfully:

data = pulumi.Output.from_input(
    {
        'values': {
            'root': {
                'cloud-config': {
                    'foo': aws_cloud.testexport.arn,
                },
            },
        },
    }
).apply(lambda x: pulumi.StringAsset(yaml.safe_dump(x)))


Environment(
    'public',
    organization=pulumi.get_organization(),
    project=pulumi.get_project(),
    name='public',
    yaml=data,
)

Nevertheless it seems to accept Input[str] for the yaml arg as well, but the type hints don't indicate that. Therefore I'd expect the wrapping into pulumi.StringAsset would be unnecessary if the type hints were extended.

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).