NF: cosmo_split to split based on feature or sample attributes
BF: ensure slice is along proper dimension