Bug fixes in dataframe_wrapper (#65629)
Summary:
## Description
- Updated functions in `dataframe_wrapper.py` to return values
- Fixed bug in `set_df_wrapper` to update `global default_wrapper`
Pull Request resolved: https://github.com/pytorch/pytorch/pull/65629
Reviewed By: ejguan
Differential Revision: D31180110
Pulled By: Nayef211
fbshipit-source-id: a8046e582fd6ce982fcdc89dae4932d0edc83d6b
