should this also use `featurize_all`? I guess that requires letting `featurize_all` support the split argument.