@Override
  protected @Nullable Presentation getPresentation(@NotNull ActionContext context, @NotNull PsiElement element) {
    if (!(context.file() instanceof PyFile)) {
      return null;
    }

    PyListCompExpression expression =
      PsiTreeUtil.getTopmostParentOfType(element, PyListCompExpression.class);
    if (expression == null) {
      return null;
    }
    if (expression.getComponents().isEmpty()) return null;
    PsiElement parent = expression.getParent();
    if (parent instanceof PyAssignmentStatement || parent instanceof PyPrintStatement) {
      return super.getPresentation(context, element);
    }
    return null;
  }
@Override
  protected @Nullable Presentation getPresentation(@NotNull ActionContext context, @NotNull PsiElement element)