A previous PR added the option to add metadata about what category id each token is. This PR forgot to update the rebalancing stage to also rebalance the category id tokens, this PR is a fix for that.
PR Checklist
[X] My PR is less than 500 lines of code
[X] I have added sufficient comment as docstrings in my code
[X] I have made corresponding changes to the documentation
[ ] I have written unit-tests to test all of my code
Summary
A previous PR added the option to add metadata about what category id each token is. This PR forgot to update the rebalancing stage to also rebalance the category id tokens, this PR is a fix for that.
PR Checklist