Fix CLI predict response JSON parsing

Author: deliahu

cli/cmd/predict.go

@@ -124,7 +124,8 @@ var predictCmd = &cobra.Command{
 				value = prediction.PredictionReversed
 			}
 
-			if casted, ok := cast.InterfaceToFloat64(value); ok {
+			if cast.IsFloatType(value) {
+				casted, _ := cast.InterfaceToFloat64(value)
 				fmt.Println(s.Round(casted, 2, true))
 			} else {
 				fmt.Println(s.UserStrStripped(value))
@@ -151,7 +152,7 @@ func makePredictRequest(apiURL string, samplesJSONPath string) (*PredictResponse
 	}
 
 	var predictResponse PredictResponse
-	err = json.Unmarshal(httpResponse, &predictResponse)
+	err = json.DecodeWithNumber(httpResponse, &predictResponse)
 	if err != nil {
 		return nil, errors.Wrap(err, "prediction response")
 	}

pkg/lib/cast/interface.go

@@ -702,6 +702,9 @@ func InterfaceToStrStrMap(in interface{}) (map[string]string, bool) {
 
 func IsIntType(in interface{}) bool {
 	switch in.(type) {
+	case json.Number:
+		_, err := in.(json.Number).Int64()
+		return err == nil
 	case int8:
 		return true
 	case int16:
@@ -718,6 +721,9 @@ func IsIntType(in interface{}) bool {
 
 func IsFloatType(in interface{}) bool {
 	switch in.(type) {
+	case json.Number:
+		_, err := in.(json.Number).Float64()
+		return err == nil
 	case float32:
 		return true
 	case float64:
@@ -731,10 +737,21 @@ func IsFloatOrIntType(in interface{}) bool {
 }
 
 func IsScalarType(in interface{}) bool {
-	if IsFloatOrIntType(in) {
-		return true
-	}
 	switch in.(type) {
+	case int8:
+		return true
+	case int16:
+		return true
+	case int32:
+		return true
+	case int64:
+		return true
+	case int:
+		return true
+	case float32:
+		return true
+	case float64:
+		return true
 	case string:
 		return true
 	case bool: