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