fix: making the uploader depend on tensorflow-proper (#499)

Co-authored-by: Yicheng Fang <yichengfang@google.com>
googleapis · Jun 28, 2021 · b95e040 · b95e040
1 parent 177aebc
commit b95e040
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/setup.py b/setup.py
@@ -29,14 +29,10 @@
 with io.open(readme_filename, encoding="utf-8") as readme_file:
     readme = readme_file.read()
 
-tensorboard_extra_require = [
-    "tensorflow-cpu>=2.3.0, <=2.5.0",
-    "grpcio~=1.38.1",
-    "six~=1.15.0",
-]
+tensorboard_extra_require = ["tensorflow >=2.3.0, <=2.5.0"]
 metadata_extra_require = ["pandas >= 1.0.0"]
 full_extra_require = tensorboard_extra_require + metadata_extra_require
-testing_extra_require = full_extra_require + ["grpcio-testing~=1.38.1"]
+testing_extra_require = full_extra_require + ["grpcio-testing"]
 
 
 setuptools.setup(