chore: refactor to remove function for simple check

Author: josetapadas

mypy/semanal.py

@@ -6178,9 +6178,7 @@ def visit_await_expr(self, expr: AwaitExpr) -> None:
             # This is not a blocker, because some enviroments (like ipython)
             # support top level awaits.
             self.fail('"await" outside function', expr, serious=True, code=codes.TOP_LEVEL_AWAIT)
-        elif not self.function_stack[-1].is_coroutine and not self.is_in_generator_expression(
-            expr
-        ):
+        elif not self.function_stack[-1].is_coroutine and self.scope_stack[-1] != SCOPE_COMPREHENSION:
             self.fail(
                 '"await" outside coroutine ("async def")',
                 expr,
@@ -6189,9 +6187,6 @@ def visit_await_expr(self, expr: AwaitExpr) -> None:
             )
         expr.expr.accept(self)
 
-    def is_in_generator_expression(self, node: AwaitExpr):
-        return self.scope_stack[-1] == SCOPE_COMPREHENSION
-
     #
     # Patterns
     #