I have not checked this code carefully, but I wonder if it was implemented from scratch or stolen from / inspired by the methods in the tree_builder script? https://github.com/JeffreyQin/Fragment-based-POMCP/blob/640ae24ed4094553327a4ec6359b62117427c4f3/generator.py#L23