Switch to unified view

a b/datasets/preprocess/statistics.py