How to we hide empty leaf in treemap plotly dash

i have 5 catergories with different depths, i would ideally want to limit the drilling of tree map based on max level of that particular catergory. For example CATEGORY A has just 3 LEVELS vs CATEGORY D has 8 levels, so ideally i would want to limit the drillng of treemap for Catergory A limited to 3 levels instead of 8 with 5 blank levels (HIDE EMPTY LEAVES).

Is there any way to achieve this, or any support that i can expect in the future