upload

package

v0.0.0-...-6f916ce Latest Latest Go to latest Published: Jun 27, 2026 License: Apache-2.0 Imports: 20 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type MockUploader struct {
	Uploader
	// contains filtered or unexported fields
}

func NewMockUploader(workers int) *MockUploader

func (u *MockUploader) Submit(ctx context.Context, resultPath string, modelVersion string)

func (u *MockUploader) Wait() error

type NgcUploader struct {
	Uploader
	// contains filtered or unexported fields
}

func NewUploader(
	ctx context.Context,
	resultsLocation, ngcBaseUrl string,
	clientSecrets *secrets.Secrets,
) (*NgcUploader, error)

func (u *NgcUploader) Submit(ctx context.Context, resultPath string, modelVersion string)

Submit an async upload job to worker.

func (u *NgcUploader) UploadResult(ctx context.Context, resultPath, version string) error

Zip result folder and upload it to NGC model registry.

func (u *NgcUploader) Wait() error

Wait for all upload worker to complete and returns error of failed uploads.

type Uploader interface {
	Submit(ctx context.Context, resultPath string, modelVersion string)
	Wait() error
}