GitLab

* Fix example

* Remove register_data_args and revert  to dataset
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>