Flickering colorbar in scatter_mapbox

[WillKoehrsen]

Thanks for the incredible library! I'm having a slight issue when I create a scatter_matbox, color the points by a continuous numerical variable, and animate by a time variable. The plot looks great except when it's animated and the legend flashes as below:

Here's the code to reproduce:

import pandas as pd
import plotly_express as px

df = pd.read_csv('https://raw.githubusercontent.com/WillKoehrsen/Data-Analysis/master/nyc_traffic_data/nyc_traffic_speed_routes_sample.csv', 
                 parse_dates=['time'])

px.set_mapbox_access_token(secrets['mapbox'])

px.scatter_mapbox(df.iloc[:10000], 
                  lat='latitude', lon='longitude', 
                  animation_frame='time', 
                  animation_group='id',
                 color='speed', title='NYC Traffic Speeds',
                 height=800, zoom=9)

This is a fairly minor issue but I was wondering if perhaps there is an simple fix. Thanks again!