Upgrade dependencies

bavard-ai · Feb 9, 2021 · 730f0bc · 730f0bc
1 parent e738373
commit 730f0bc
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/dockerfiles/test.Dockerfile b/dockerfiles/test.Dockerfile
@@ -1,4 +1,4 @@
-FROM tensorflow/tensorflow:2.2.2
+FROM tensorflow/tensorflow:2.3.2
 
 WORKDIR /app
 

diff --git a/requirements-test.txt b/requirements-test.txt
@@ -2,5 +2,6 @@ requests==2.24.0
 flake8==3.8.4
 scikit-learn==0.23.2
 torch==1.7.0
-transformers==3.4.0
-tensorflow==2.2.2
+transformers==4.3.0
+tensorflow==2.3.2
+sentencepiece==0.1.95
diff --git a/setup.py b/setup.py
@@ -31,7 +31,7 @@ def run(self):
 setuptools.setup(
     name="bavard-ml-common",
     version=VERSION,
-    author="Bavard AI, LLC",
+    author="Bavard AI, Inc.",
     author_email="[email protected]",
     description="Machine learning model utilities",
     long_description=long_description,
@@ -49,7 +49,6 @@ def run(self):
         "pydantic==1.7.3",
         "google-cloud-storage==1.35.1",
         "google-cloud-pubsub==2.2.0",
-        "protobuf==3.12.0",  # needed b/c of: https://github.com/googleapis/python-bigquery/issues/305
         "numpy>=1.15.4",
         "scikit-learn>=0.20.3"
     ],