Remove tensorflow and tensorflow-datasets as explicit install requirements.

Both packages are still required to use all functionalities of `clu`. However there are several packages providing TF (default, nightly, macos) and we should not force a specific version (see github.com/google/seqio/issues/396).

Also bump version to 0.0.8.

PiperOrigin-RevId: 495878761

Author: Marvin182

setup.py

@@ -24,6 +24,8 @@
 
 tests_require = [
     "pytest",
+    "tensorflow",
+    "tensorflow_datasets",
     "torch>=1.2.0",
 ]
 pytorch_require = [
@@ -32,7 +34,7 @@
 
 setup(
     name="clu",
-    version="0.0.7",
+    version="0.0.8",
     description=("Set of libraries for ML training loops in JAX."),
     author="Common Loop Utils Authors",
     author_email="no-reply@google.com",
@@ -51,9 +53,8 @@
         "ml_collections",
         "numpy",
         "packaging",
-        "tensorflow",
-        "tensorflow_datasets",
         "typing_extensions",
+        "wrapt",
     ],
     tests_require=tests_require,
     extras_require=dict(test=tests_require, pytorch=pytorch_require),