Refactoring for better tracking of the tile coding code
Improve robustness for WrappedPartition