@Override
  protected boolean isApplicableElement(@NotNull PsiElement element) {
    return element instanceof PyFunction || element instanceof PyClass || element instanceof PyFile;
  }
@Override
  protected boolean isApplicableElement(@NotNull PsiElement element)