@pprett actually I was also unsure here. I realized this probable wasn't needed but was following what has been done in the tree module. See `sklearn/tree/tree.py` line 401 etc.