CR feedback. Return explicit bid type. (#2)

prebid · Nov 16, 2023 · c68be34 · c68be34
1 parent 14ebf6b
commit c68be34
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/adapters/yieldmo/yieldmo.go b/adapters/yieldmo/yieldmo.go
@@ -171,5 +171,14 @@ func getMediaTypeForImp(bid openrtb2.Bid) (openrtb_ext.BidType, error) {
 		return "", &errortypes.BadInput{Message: err.Error()}
 	}
 
-	return openrtb_ext.ParseBidType(bidExt.MediaType)
+	switch bidExt.MediaType {
+	case "banner":
+		return openrtb_ext.BidTypeBanner, nil
+	case "video":
+		return openrtb_ext.BidTypeVideo, nil
+	case "native":
+		return openrtb_ext.BidTypeNative, nil
+	default:
+		return "", fmt.Errorf("invalid BidType: %s", bidExt.MediaType)
+	}
 }