Plotly.express.scatter marginal_y and color

I have a scatter plot using express which is colored by a categorical column. The main scatter plot is colored. However, marginal_y plot is not colored. It will be good if the marginal_y sub-plot is also colored, either by default when color is enabled or through a new argument.