checkpoint: adapting downstream code like ShuffleSplit() and unittests to new Split() signature