Missing code for using train_aug_data.json in env_mapdataset.py

[KlrShaK]

Hello, while trying to reproduce results for the repository I noticed that your scripts are lacking the functionality to use your LLM-augmented instructions dataset (train_aug_data.json). Even after replacing the original train_data.json file with your augmented file I receive an error due to the formatting difference between your file and the original (namely your file have multiple instructions inside pre_dialogs and instructions) which results in an error in the following line number 288 and 289.

item['instructions'] = item['instructions'].lower()
item['pre_dialogs'] = ' '.join(item['pre_dialogs']).lower()

I please request you to provide your code file for reproducibility