@Override
    protected boolean isInContext(@NotNull PsiElement element) {
      return PsiTreeUtil.getParentOfType(element, PyClass.class) != null;
    }
@Override
    protected boolean isInContext(@NotNull PsiElement element)