老师您好，按照您给的源码，我存到了谷歌的colab运行，text-rnn.ipynb报错，自己弄了好久还是没解决，请教下。
代码报错提示为：

ExceptionTraceback (most recent call last)
in ()
58
59 train_dataset = TextDataSet(
—> 60 train_file, vocab, category_vocab, hps.num_timesteps)
61 val_dataset = TextDataSet(
62 val_file, vocab, category_vocab, hps.num_timesteps)
2 frames
in category_to_id(self, category)
55 if not category in self._category_to_id:
56 raise Exception(
—> 57 “%s is not in our category list” % category)
58 return self._category_to_id[category]
59

代码运行报错的部分为：

def category_to_id(self, category):

    if not category in self._category_to_id:
        raise  Exception( 
            "%s is not in our category list" % category)
    return self._category_to_id[category]

整个代码的可在线查看链接为：https://drive.google.com/file/d/1dZOvfOHn40gN06Pu1HnaN37xdtZC3pVR/view?usp=sharing