The package requirements and setup.py should indicate that it is compatible tensorflow-datasets==1.2.0.