@Override
  public void collectElementsToIgnore(@NotNull PsiElement element, @NotNull Set<PsiElement> result) {
    PsiElement parent = element.getParent();

    while (parent instanceof PyIfStatement) {
      result.add(parent);
      parent = parent.getParent();
    }
  }
@Override
  public void collectElementsToIgnore(@NotNull PsiElement element, @NotNull Set<PsiElement> result)