Handle cx predict intermediate API unavailable state

cortexlabs · Feb 9, 2019 · c45619fa32bf042c73637f540c925c53fe63b2ca · c45619f
1 parent 471704e
commit c45619fa32bf042c73637f540c925c53fe63b2ca
Unified Split

Showing with 1 addition and 1 deletion.

+1 −1 cli/cmd/predict.go
diff --git a/cli/cmd/predict.go b/cli/cmd/predict.go
@@ -80,7 +80,7 @@ var predictCmd = &cobra.Command{
 		apiURL := util.URLJoin(resourcesRes.APIsBaseURL, apiPath)
 		predictResponse, err := makePredictRequest(apiURL, samplesJSONPath)
 		if err != nil {
-  
-			if strings.Contains(err.Error(), "503 Service Temporarily Unavailable") {
+  
+			if strings.Contains(err.Error(), "503 Service Temporarily Unavailable") || strings.Contains(err.Error(), "502 Bad Gateway") {
 				errors.Exit(s.ErrApiNotReady(apiName, resource.StatusAPIUpdating.Message()))
 			}
 			errors.Exit(err)