Getting marginals for nested PartitionModeState
The PartitionModeState functions allow to get the 2D matrix of marginals given the models used to create the object. When it is built using NestedBlockState instances, the marginals are only returned for the deepest BlockState in the hierarchy. Would it be possible to implement a strategy to get the marginals of the blocks at higher levels (i.e. the marginals for blocks at level x to belong to blocks at level x + 1)?