PsiElement (@NotNull PsiElement element, @Nullable Editor editor) { final PyFile file = (PyFile) element; if (file.getName().equals(PyNames.INIT_DOT_PY) && editor != null) { return file.getParent(); } return element; }