Switch to tfds-nightly

tfboyd · web-flow · commit 5b347e5bd877 · 2019-02-20T18:42:20.000-08:00
diff --git a/perfzero/docker/Dockerfile_ubuntu_1804_tf_v2 b/perfzero/docker/Dockerfile_ubuntu_1804_tf_v2
@@ -65,8 +65,7 @@ RUN pip3 install --upgrade pip==9.0.1
 RUN pip3 install --upgrade setuptools google-api-python-client pyyaml google-cloud google-cloud-bigquery
 RUN pip3 install wheel absl-py
 RUN pip3 install --upgrade --force-reinstall tf-nightly-gpu-2.0-preview
-RUN pip3 install tensorflow-datasets
-RUN pip3 install tensorflow-metadata
+RUN pip3 install tfds-nightly
 
 RUN curl https://raw.githubusercontent.com/tensorflow/models/master/official/requirements.txt > /tmp/requirements.txt
 RUN pip3 install -r /tmp/requirements.txt
