Add hook in FPS to allow transform hooks
Fix bug where probability of 0 gets picked
Add MockSampler class
Add test and hooks for FPS transforms