tfds.load to GCS hangs

[martin-gorner]

Repro snippet: ! Please use your own writable GCS bucket in place of "your_bucket" tfds.load(name="caltech_birds2011", shuffle_files=True, data_dir="gs://your_bucket")

Hangs indefinitely with "Extraction completed...: 0% 0/2 [00:21<?, ? file/s]"

• Operating System: Linux (Vertex AI Workbench default)
• Python version: 3.6
• tensorflow-datasets/tfds-nightly version: 4.6.0
• tensorflow/tf-nightly version: 2.9.1
• Does the issue still exists with the last tfds-nightly package (pip install --upgrade tfds-nightly) YES