Different color for markers when using go.Scattermapbox

Hello,

In px.pscatter_mapbox(), if I specify the name of a dataframe column for color attribute, I get points on a map with two different colors (image attached for reference). How can I replicate the same when using go.scattermapbox?