Skip to content

Commit 45c7433

Browse files
deliahuvishalbollu
authored andcommitted
Add error wrapping for predictor model key (#867)
(cherry picked from commit 7460edf)
1 parent 69180ae commit 45c7433

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

pkg/operator/operator/validations.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -492,7 +492,7 @@ func validateTensorFlowPredictor(predictor *userconfig.Predictor) error {
492492

493493
awsClient, err := aws.NewFromEnvS3Path(model)
494494
if err != nil {
495-
return err
495+
return errors.Wrap(err, userconfig.ModelKey)
496496
}
497497

498498
if strings.HasSuffix(model, ".zip") {
@@ -502,7 +502,7 @@ func validateTensorFlowPredictor(predictor *userconfig.Predictor) error {
502502
} else {
503503
path, err := getTFServingExportFromS3Path(model, awsClient)
504504
if err != nil {
505-
return err
505+
return errors.Wrap(err, userconfig.ModelKey)
506506
} else if path == "" {
507507
return errors.Wrap(ErrorInvalidTensorFlowDir(model), userconfig.ModelKey)
508508
}
@@ -521,7 +521,7 @@ func validateONNXPredictor(predictor *userconfig.Predictor) error {
521521

522522
awsClient, err := aws.NewFromEnvS3Path(model)
523523
if err != nil {
524-
return err
524+
return errors.Wrap(err, userconfig.ModelKey)
525525
}
526526

527527
if ok, err := awsClient.IsS3PathFile(model); err != nil || !ok {

0 commit comments

Comments
 (0)