transformers

de46cde1 - Drop columns after loading samples in prepare_tf_dataset (#17967)

Drop columns after loading samples in prepare_tf_dataset (#17967) * Drop columns after loading samples, rather than before, to avoid breaking transforms * make fixup * Add workaround so this PR can work with current datasets version