Add test coverage for data_transformation.py and fix DeprecationWarning bug#274

Copilot wants to merge 5 commits intomainfacebookresearch/balance:mainfrom

copilot/add-missing-tests-balance-utilsfacebookresearch/balance:copilot/add-missing-tests-balance-utilsCopy head branch name to clipboard