use auth to avoid docker hub rate limits

michaeldwan · michaeldwan · commit de2a62ed81a6 · 2025-07-03T17:16:31.000-06:00
diff --git a/tools/compatgen/internal/cuda.go b/tools/compatgen/internal/cuda.go
@@ -8,6 +8,7 @@ import (
 	"strings"
 
 	"github.com/anaskhan96/soup"
+	"github.com/google/go-containerregistry/pkg/authn"
 	"github.com/google/go-containerregistry/pkg/name"
 	"github.com/google/go-containerregistry/pkg/v1/remote"
 	"golang.org/x/sync/errgroup"
@@ -99,7 +100,7 @@ func parseCUDABaseImage(ctx context.Context, tag string) (*config.CUDABaseImage,
 		return nil, fmt.Errorf("Failed to parse reference %s: %w", tag, err)
 	}
 
-	img, err := remote.Image(ref, remote.WithContext(ctx))
+	img, err := remote.Image(ref, remote.WithContext(ctx), remote.WithAuthFromKeychain(authn.DefaultKeychain))
 	if err != nil {
 		return nil, fmt.Errorf("Failed to get image %s: %w", tag, err)
 	}

Original file line number	Diff line number	Diff line change
`@@ -8,6 +8,7 @@ import (`
`8`	`8`	`"strings"`
`9`	`9`
`10`	`10`	`"github.com/anaskhan96/soup"`
	`11`	`+ "github.com/google/go-containerregistry/pkg/authn"`
`11`	`12`	`"github.com/google/go-containerregistry/pkg/name"`
`12`	`13`	`"github.com/google/go-containerregistry/pkg/v1/remote"`
`13`	`14`	`"golang.org/x/sync/errgroup"`
`@@ -99,7 +100,7 @@ func parseCUDABaseImage(ctx context.Context, tag string) (*config.CUDABaseImage,`
`99`	`100`	`return nil, fmt.Errorf("Failed to parse reference %s: %w", tag, err)`
`100`	`101`	`}`
`101`	`102`
`102`		`- img, err := remote.Image(ref, remote.WithContext(ctx))`
	`103`	`+ img, err := remote.Image(ref, remote.WithContext(ctx), remote.WithAuthFromKeychain(authn.DefaultKeychain))`
`103`	`104`	`if err != nil {`
`104`	`105`	`return nil, fmt.Errorf("Failed to get image %s: %w", tag, err)`
`105`	`106`	`}`