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

yfang1 · Yicheng Fang · web-flow · commit b95e04065668 · 2021-06-28T15:00:07.000-04:00
Co-authored-by: Yicheng Fang &lt;yichengfang@google.com&gt;
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(
