fix: try with the definition context

Author: bzoracler

mypy/checker.py

@@ -1223,14 +1223,11 @@ def check_func_def(
                     if defn.is_async_generator:
                         if not self.is_async_generator_return_type(typ.ret_type):
                             self.fail(
-                                message_registry.INVALID_RETURN_TYPE_FOR_ASYNC_GENERATOR,
-                                typ.ret_type,
+                                message_registry.INVALID_RETURN_TYPE_FOR_ASYNC_GENERATOR, defn
                             )
                     else:
                         if not self.is_generator_return_type(typ.ret_type, defn.is_coroutine):
-                            self.fail(
-                                message_registry.INVALID_RETURN_TYPE_FOR_GENERATOR, typ.ret_type
-                            )
+                            self.fail(message_registry.INVALID_RETURN_TYPE_FOR_GENERATOR, defn)
 
                 # Fix the type if decorated with `@types.coroutine` or `@asyncio.coroutine`.
                 if defn.is_awaitable_coroutine: