@Override
  protected @Nullable Presentation getPresentation(@NotNull ActionContext context, @NotNull PyForStatement element) {
    if (!LanguageLevel.forElement(element).isPython2()) {
      final PyTargetExpression forTarget = findSingleForLoopTarget(element);
      final PyReferenceExpression yieldValue = findSingleYieldValue(element);
      if (forTarget != null && yieldValue != null) {
        final String targetName = forTarget.getName();
        if (targetName != null && targetName.equals(yieldValue.getName())) {
          return super.getPresentation(context, element);
        }
      }
    }
    return null;
  }
@Override
  protected @Nullable Presentation getPresentation(@NotNull ActionContext context, @NotNull PyForStatement element)