Isolate TF dependencies to lit_nlp.examples module.

RyanMullins · LIT team · commit 2138bd920e72 · 2024-01-31T11:22:26.000-08:00
PiperOrigin-RevId: 603107736
diff --git a/lit_nlp/examples/models/tfx_model.py b/lit_nlp/examples/models/tfx_model.py
diff --git a/lit_nlp/examples/models/tfx_model_test.py b/lit_nlp/examples/models/tfx_model_test.py
@@ -2,7 +2,7 @@
 import tempfile
 
 from lit_nlp.api import types as lit_types
-from lit_nlp.components import tfx_model
+from lit_nlp.examples.models import tfx_model
 import tensorflow as tf
 
 
diff --git a/pyproject.toml b/pyproject.toml
@@ -36,8 +36,6 @@ dependencies = [
   "scipy==1.10.1",
   "shap==0.37.0",
   "six==1.16.0",
-  "tensorflow==2.10.0",
-  "tensorflow-text==2.10.0",
   "termcolor==2.3.0",
   "tqdm==4.64.0",
   "umap-learn==0.5.1",
@@ -82,7 +80,9 @@ keywords = [
 # LINT.IfChange
 examples = [
   "gunicorn==20.1.0",
+  "tensorflow==2.10.0",
   "tensorflow-datasets==4.8.0",
+  "tensorflow-text==2.10.0",
   "torch==2.0.1",
   "transformers==4.27.1",
 ]
diff --git a/requirements_core.txt b/requirements_core.txt
@@ -35,8 +35,6 @@ scikit-learn==1.0.2
 scipy==1.10.1
 shap==0.37.0
 six==1.16.0
-tensorflow==2.10.0
-tensorflow-text==2.10.0
 termcolor==2.3.0
 tqdm==4.64.0
 umap-learn==0.5.1
diff --git a/requirements_examples.txt b/requirements_examples.txt
@@ -15,7 +15,9 @@
 # LINT.IfChange
 gunicorn==20.1.0
 sentencepiece==0.1.99
+tensorflow==2.10.0
 tensorflow-datasets==4.8.0
+tensorflow-text==2.10.0
 torch==2.0.1
 transformers==4.27.1
 # LINT.ThenChange(./pyproject.toml)
