@Override
  protected boolean isCustomFoldingRoot(@NotNull ASTNode node) {
    return node.getPsi() instanceof PyFile || node.getElementType() == PyElementTypes.STATEMENT_LIST;
  }
@Override
  protected boolean isCustomFoldingRoot(@NotNull ASTNode node)