From b95e0406566879e8f71cefda72b41dc6fe4e578f Mon Sep 17 00:00:00 2001 From: Yicheng Fang <58752348+yfang1@users.noreply.github.com> Date: Mon, 28 Jun 2021 12:00:07 -0700 Subject: [PATCH] fix: making the uploader depend on tensorflow-proper (#499) Co-authored-by: Yicheng Fang --- setup.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/setup.py b/setup.py index b883ae841d..d59e9f8cb1 100644 --- 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(