added the option to return the run and traj idx
added the option to return the run and traj idx

This is needed to know where the data came from if you don't want to
parse the path to the dataset. WHich I didn't want to do.

ADicksonLab/wepy,ADicksonLab/wepy,ADicksonLab/wepy