Cannot load data from imdb_reviews datasets.

Please make sure that this is a bug. As per our
GitHub Policy,
we only address code/doc bugs, performance issues, feature requests and
build/installation issues on GitHub. tag:bug_template

System information

• Have I written custom code (as opposed to using a stock example script provided in TensorFlow):
• OS Platform and Distribution (e.g., Linux Ubuntu 16.04): Ubuntu20.04
• Mobile device (e.g. iPhone 8, Pixel 2, Samsung Galaxy) if the issue happens on mobile device:
• TensorFlow installed from (source or binary): conda
• TensorFlow version (use command below): 2.2.0
• Python version: 3.8.3
• Bazel version (if compiling from source):
• GCC/Compiler version (if compiling from source):
• CUDA/cuDNN version: 10.1, 7.6.5
• GPU model and memory: RTX2060, VRAM6GB

You can collect some of this information using our environment capture
script
You can also obtain the TensorFlow version with:

1. TF 1.0: python -c "import tensorflow as tf; print(tf.GIT_VERSION, tf.VERSION)"
2. TF 2.0: python -c "import tensorflow as tf; print(tf.version.GIT_VERSION, tf.version.VERSION)"

Describe the current behavior
When I run

import tensorflow as tf
import tensorflow_datasets as tfds

imdb, info = tfds.load('imdb_reviews', with_info=True, as_supervised=True)

the only output is

Downloading and preparing dataset imdb_reviews (80.23 MiB) to /root/tensorflow_datasets/imdb_reviews/plain_text/0.1.0...
HBox(children=(FloatProgress(value=1.0, bar_style='info', description='Dl Completed...', max=1.0, style=Progre…
HBox(children=(FloatProgress(value=1.0, bar_style='info', description='Dl Size...', max=1.0, style=ProgressSty…

and loading takes forever to end(won't end).

I tried to import data from keras.datasets and there seems no problems with it(was able to get the data imported with no warning and error).
But the problem is I don't know how to import subwords version using keras.datasets as below;

dataset, info = tfds.load('imdb_reviews/subwords8k', with_info=True, as_supervised=True)

I also show the details of my environment(I'm using docker)

FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04
RUN apt-get update && apt-get install -y
sudo
wget
vim
WORKDIR /opt
RUN wget https://repo.continuum.io/archive/Anaconda3-2020.07-Linux-x86_64.sh &&
sh Anaconda3-2020.07-Linux-x86_64.sh -b -p /opt/anaconda3 &&
rm -f Anaconda3-2020.07-Linux-x86_64.sh

ENV PATH /opt/anaconda3/bin:$PATH

RUN conda update conda && conda install
keras
scipy
tensorflow-gpu

WORKDIR /

CMD ["jupyter", "lab", "--ip=0.0.0.0", "--allow-root", "--LabApp.tokenh=''"]

Describe the expected behavior

Standalone code to reproduce the issue
Provide a reproducible test case that is the bare minimum necessary to generate
the problem. If possible, please share a link to Colab/Jupyter/any notebook.

Other info / logs Include any logs or source code that would be helpful to
diagnose the problem. If including tracebacks, please include the full
traceback. Large logs and files should be attached.

Start by reproducing the tfds.load('imdb_reviews', with_info=True, as_supervised=True) call in the reported Ubuntu, Python, TensorFlow, CUDA, and Docker environment. Compare it with the imdb_reviews/subwords8k call and investigate where dataset preparation stops; done means the loading behavior and cause are identified with supporting logs or a reproducible test.