Sample size not exact for sample_usability.py

Due to rounding from the way stratified sampling is implemented currently, the number of samples in the output may be different from the sample size requested. Either we should change the implementation or keep it but make clear to users that is the behavior.