TfdsDataProvider gives error with non-None tfds_data_dir about seqio HOT 2 CLOSED

import os
import seqio
import functools

task_name = "wiki40b_en"
data_dir = os.path.abspath(os.path.expanduser("~/my_tfds_test_dir"))
tfds_id = "wiki40b/en:1.3.0"
tfds_col = "text"
vocab = seqio.ByteVocabulary()  # whatever

seqio.TaskRegistry.add(
    task_name,
    source=seqio.TfdsDataSource(tfds_id, tfds_data_dir=data_dir),
    preprocessors=[
        functools.partial(seqio.preprocessors.rekey, key_map={"targets": tfds_col}),
        seqio.preprocessors.tokenize,
    ],
    output_features={"targets": seqio.Feature(vocabulary=vocab)},
)
ds = seqio.TaskRegistry.get(task_name).get_dataset(sequence_length=None, split="train", shuffle=False)