PsiElement (@NotNull ASTNode node) { final IElementType type = node.getElementType(); if (type instanceof PyElementType pyElType) { return pyElType.createElement(node); } else if (type instanceof PyStubElementType) { return ((PyStubElementType<?, ?>)type).createElement(node); } return new ASTWrapperPsiElement(node); }