Install TF Text on non-Windows only (#2115)

keras-team · Feb 27, 2025 · 4801985 · 4801985
1 parent c3805d3
commit 4801985
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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",