diff --git a/lit_nlp/components/tfx_model.py b/lit_nlp/examples/models/tfx_model.py similarity index 100% rename from lit_nlp/components/tfx_model.py rename to lit_nlp/examples/models/tfx_model.py diff --git a/lit_nlp/components/tfx_model_test.py b/lit_nlp/examples/models/tfx_model_test.py similarity index 97% rename from lit_nlp/components/tfx_model_test.py rename to lit_nlp/examples/models/tfx_model_test.py index 31dcf628..4b44b2d4 100644 --- a/lit_nlp/components/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 index 3c8bb401..e3ccdde7 100644 --- 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 index e3fe7551..98a4b651 100644 --- 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 index d3353493..c19c4e1b 100644 --- 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)