@NotNull
  private static StreamEx<PsiElement> expandResolveAugAssignments(@NotNull PsiElement element,
                                                                  @NotNull PyResolveContext context,
                                                                  @NotNull Set<PyReferenceOwner> alreadyVisited) {
    if (element instanceof PyReferenceExpression && PyAugAssignmentStatementNavigator.getStatementByTarget(element) != null) {
      return StreamEx.of(resolveReferenceAugAssignmentsAware((PyReferenceOwner)element, context, alreadyVisited));
    }
    return StreamEx.of(element);
  }
@NotNull
  private static StreamEx<PsiElement> expandResolveAugAssignments(@NotNull PsiElement element,
                                                                  @NotNull PyResolveContext context,
                                                                  @NotNull Set<PyReferenceOwner> alreadyVisited)