From b9ad489137f1662fba5fc18b99e5759a44acc721 Mon Sep 17 00:00:00 2001
From: Abheesht Sharma <abheesht@google.com>
Date: Thu, 27 Feb 2025 11:31:52 +0530
Subject: [PATCH] Install TF Text on non-Windows only

---
 requirements.txt | 2 +-
 setup.py         | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 552b1ec299..7b59e73759 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,7 @@
 # Tensorflow.
 tensorflow-cpu~=2.18.0;sys_platform != 'darwin'
 tensorflow~=2.18.0;sys_platform == 'darwin'
-tensorflow-text~=2.18
+tensorflow-text~=2.18;platform_system != 'Windows'
 
 # Torch.
 --extra-index-url https://download.pytorch.org/whl/cpu
diff --git a/setup.py b/setup.py
index b9f1588039..138e295af8 100644
--- a/setup.py
+++ b/setup.py
@@ -45,13 +45,13 @@ def get_version(rel_path):
         "regex",
         "rich",
         "kagglehub",
+        "tensorflow-text;platform_system != 'Windows'",
     ],
     extras_require={
         "extras": [
             "rouge-score",
             "sentencepiece",
         ],
-        "nlp": ["tensorflow-text"],
     },
     # Supported Python versions
     python_requires=">=3.9",