private static boolean shouldCollectFeatures(@NotNull PsiBlockStatement statement) {
    PsiElement parentStatement = statement.getParent();
    return statement.getCodeBlock().getStatementCount() <= 1 ||
           !(parentStatement instanceof PsiLoopStatement || parentStatement instanceof PsiIfStatement);
  }
private static boolean shouldCollectFeatures(@NotNull PsiBlockStatement statement)