tfds.loadでAttributeErrorが出た

tfds.loadで、AttributeErrorが出たので対処法のメモです。

環境

tensorflow 2.0.0-rc1
tensorflow-datasets-1.0.1

起こったこと

いい加減MNISTをやめてほかのデータセットを使おうと思い、fashion_mnistをロードしようとしたところ、エラーが発生しました。

AttributeError: '_OptionsDataset' object has no attribute 'output_shapes'

解決した方法

tensorflowのissueに同じ症状が報告されていました。
github.com

tensorflow-datasetsのバージョンが古いと起こるようなのでアップグレードすることで解決しました。

pip install --upgrade tensorflow-datasets