Incorrect type on .result() for a submitted async task results in invalid mypy suggestion

[mthanded]

Bug summary

If you call .result() on a submitted async task mypy believes that the call should be awaited. However the .result() is not a coroutine and following the mypy suggestion results in a TypeError.

Example:

from prefect import task, flow
import asyncio


@task
async def test_task() -> int:
    return 1

def test_function(a:int) -> int:
    return a

@flow 
async def test_flow() -> None:
    a = test_task.submit().result()
    print(type(a))
    b = test_function(a)
    print(b)

if __name__ == "__main__" :
    asyncio.run(test_flow())

The code above runs correctly outputting

<class 'int'>
1

However mypy thinks that line 16 ,b = test_function(a), has a problem

test.py:16: error: Argument 1 to "test_function" has incompatible type "Coroutine[Any, Any, int]"; expected "int"  [arg-type]
test.py:16: note: Maybe you forgot to use "await"?

If you await .result() you get:

TypeError: object int can't be used in 'await' expression

Version info

mypy   1.14.1    
prefect  3.1.13 | 3.1.14.dev3

Additional context

No response

[zzstoatzz]

thanks for the issue @mthanded - I think this is fundamentally the same issue as #15008. I think it's worth tracking this independently though