Dynamically adjust continuous color range in treemap as I drill down

Hi, I’m using the color= parameter in px.treemap which works as expected at the root level. However, with maxdepth=3, when I drill into the sections, the color range doesn’t change. This is a problem since the absolute values of the children would likely be at orders of magnitude smaller than root, resulting in basically no color differences across the sections after drilling down from the root.

Is there a workaround for this? Thanks!