ImportError: No module named 'cPickle'， however, cPickle is obsolete。。。

[SeekPoint]

$t2t-datagen --data_dir=$DATA_DIR --tmp_dir=$TMP_DIR --num_shards=100 --problem=$PROBLEM
Traceback (most recent call last):
File "/tf1.2py3_venv/venv/bin/t2t-datagen", line 39, in
from tensor2tensor.data_generators import image
File "/tf1.2py3_venv/venv/lib/python3.6/site-packages/tensor2tensor/data_generators/image.py", line 21, in
import cPickle
ImportError: No module named 'cPickle'

[lukaszkaiser]

Thanks for pointing this out! Would you like to help and make a PR that solves it?

[SeekPoint]

I had this issue solved, but I think maybe #2 is also on going...