Fix installation (#1254)

Fixes #1253

Author: LukeWood

setup.py

@@ -49,15 +49,14 @@ def is_pure(self):
     author="Keras team",
     author_email="[email protected]",
     license="Apache License 2.0",
-    install_requires=["packaging", "absl-py", "regex"],
+    install_requires=["packaging", "absl-py", "regex", "tensorflow-datasets"],
     python_requires=">=3.7",
     extras_require={
         "tests": [
             "flake8",
             "isort",
             "black[jupyter]",
             "pytest",
-            "tensorflow-datasets",
             "pycocotools",
         ],
         "examples": ["tensorflow_datasets", "matplotlib"],