Rebasing

kserve · Oct 13, 2020 · ddc2ffb · ddc2ffb
1 parent b98b51d
commit ddc2ffb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cmd/agent/main.go b/cmd/agent/main.go
@@ -37,7 +37,7 @@ func main() {
 			panic(err)
 		}
 		sessionClient := s3.New(sess)
-		downloader.Providers[kfstorage.S3] = &storage.S3Provider{
+		downloader.Providers[kfstorage.S3] = &kfstorage.S3Provider{
 			Client: sessionClient,
 			Downloader: s3manager.NewDownloaderWithClient(sessionClient, func(d *s3manager.Downloader) {
 			}),

diff --git a/pkg/agent/kfstorage/gcs.go b/pkg/agent/kfstorage/gcs.go
@@ -18,7 +18,7 @@ type GCSProvider struct {
 	Client mockapi.Client
 }
 
-func (p *GCSProvider) Download(modelDir string, modelName string, storageUri string) error {
+func (p *GCSProvider) DownloadModel(modelDir string, modelName string, storageUri string) error {
 	gcsUri := strings.TrimPrefix(storageUri, string(GCS))
 	path := strings.Split(gcsUri, "/")
 	ctx := context.Background()