Icon (@NotNull PsiElement element, int flags) { if (element instanceof PsiDirectory directory) { // Preserve original icons for excluded directories and source roots if (isSpecialDirectory(directory)) return null; if (PyUtil.isExplicitPackage(directory)) { return AllIcons.Nodes.Package; } } return null; }