Hi Prefect, I started seeing the following typing errors when checking my code with `mypy`:

error: Signature of "run" incompatible with supertype "Task"

For example for following Task:

class RestTaskTest(prefect.Task):
    def run(self, string_val: str, int_val: int) -> TaskTestResult:
        return TaskTestResult(string_val=string_val, int_val=int_val)

This makes sense since the definition of

Task.run

is

def run(self) -> None:

. Is there any way to make mypy work together with overriding of Task.run in custom tasks? Or is the only option to ignore these with

# type: ignore

?

Hey again , could you open an enhancement request, potentially to change the annotation for Task.run to

Any

?

You are right, the return type should be more general by being

Any

. However, I don't think its only the Any. I tried changing that in prefect, and the error is still there. There are also the Task method arguments that are overwritten

self

is overwritten by

self, string_val: str, int_val: int

in the previous example. And I don't know how to correctly type this myself.

if you decorate run with @defaults_from_attrs it’ll make mypy happy, thats how i got around that

Thanks @Zach Schumacher, using

@defaults_from_attrs()

on top of the overridden

run()

method works. Thanks!