Mapping US Census Tracts

Is there a way to map Census Tracts instead of Counties? Here is an example of plotting by county: Choropleth Maps | Python | Plotly

I would like to create the same map by tract.

Thank you!