private @Nullable PyType getTypeForImports(@NotNull PyFunction function, @NotNull TypeEvalContext context) {
    PyType returnTypeActual = getActualReturnType(function, myReturnExpr != null ? myReturnExpr.getElement() : null, context);
    if (myHaveSuggestedType && returnTypeActual != null) {
      return returnTypeActual;
    }
    else {
      return function.getReturnStatementType(context);
    }
  }
private @Nullable PyType getTypeForImports(@NotNull PyFunction function, @NotNull TypeEvalContext context)