public static boolean isPyiFileOfPackage(@NotNull PsiElement element) {
    return element instanceof PyiFile || PyUtil.turnDirIntoInitPyi(element) instanceof PyiFile;
  }
public static boolean isPyiFileOfPackage(@NotNull PsiElement element)