@Nullable
  private static PyType getReturnType(@Nullable PsiElement element, @NotNull TypeEvalContext context) {
    if (element instanceof PyTypedElement) {
      final PyType type = context.getType((PyTypedElement)element);

      return getReturnType(type, context);
    }

    return null;
  }
@Nullable
  private static PyType getReturnType(@Nullable PsiElement element, @NotNull TypeEvalContext context)