Missing dataset.root parameter in the training script train.py

[rkeshwani]

When you record your datasets, you are given the option to specify the root directory where your training data will be stored. BUT when you go to train your policy, you are not given the ability to specify the location of where your data root is located. It defaults to the user home directory.

So you cannot train policies on your local data if you have changed your root directory.

If I'm missing something, please let me know!