Adds tensorflow-datasets for both TF 2.17 and TF 2.15 environments

requirements.tf-2.15.0.txt

@@ -1,6 +1,7 @@
 tensorflow-probability<=0.23.0
 tensorflow-text<=2.15.0 ; platform_machine != "aarch64" and platform_machine != "arm64"
 keras-nlp==0.6.3 ; platform_machine != "aarch64"
+tensorflow-datasets
 # At this time, the current release is 4.45.1.
 transformers<5.0.0
 -r common-requirements.txt

requirements.tf-2.17.0.txt

@@ -1,6 +1,7 @@
 tensorflow-probability[tf]<=0.24.0
 tensorflow-metal ; platform_machine == "arm64" and sys_platform == 'darwin'
 tensorflow-text==2.17.0 ; platform_machine != "aarch64"
+tensorflow-datasets
 keras-nlp ; platform_machine != "aarch64"
 tf-keras
 # As per its own req's, transformers needs TF<2.16, but apparently it