Make dependency on tensorflow-compression explicit.

fredrec · tensorflower-gardener · commit 4bcbdd511077 · 2022-03-18T00:50:22.000-07:00
PiperOrigin-RevId: 435565408
diff --git a/ci/kokoro/run_bazel_unittests.sh b/ci/kokoro/run_bazel_unittests.sh
@@ -33,8 +33,6 @@ set -o pipefail  # Treat the failure of a command in a pipeline as error.
 # set -x
 
 pip install --requirement "requirements.txt"
-# Not in list of requirements, but needed for EPR test:
-pip install tensorflow-compression
 
 # Run the tests.
 # Some tests requiring more RAM that the CI machine provides are disabled.
diff --git a/requirements.txt b/requirements.txt
@@ -5,3 +5,4 @@ scipy
 enum34~=1.1
 mock
 dm-tree~=0.1.1
+tensorflow-compression~=2.8.1
