From 70603d03e2e698150f6608fafe615e1190d13e4e Mon Sep 17 00:00:00 2001
From: Hilmar Lapp <hlapp@drycafe.net>
Date: Wed, 27 Nov 2024 00:13:54 -0500
Subject: [PATCH] Adds tensorflow-datasets for both TF 2.17 and TF 2.15
 environments

---
 requirements.tf-2.15.0.txt | 1 +
 requirements.tf-2.17.0.txt | 1 +
 2 files changed, 2 insertions(+)

diff --git a/requirements.tf-2.15.0.txt b/requirements.tf-2.15.0.txt
index ff8933b..c5d5ab8 100644
--- a/requirements.tf-2.15.0.txt
+++ b/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
diff --git a/requirements.tf-2.17.0.txt b/requirements.tf-2.17.0.txt
index d5369b0..b500553 100644
--- a/requirements.tf-2.17.0.txt
+++ b/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