def load_hred_vocabulary(args):
path = os.path.join(args.data_dir, 'experiment_data', 'hred', 'words.pkl')
with open(path, 'rb') as words_file:
words = pkl.load(words_file)
word_dict = {word: idx for idx, word in enumerate(words.keys())}
logging.debug(type(words))
vocab = Vocabulary.from_dict(word_dict)
return vocab
