Make dependency on tensorflow-compression explicit.

PiperOrigin-RevId: 437685807

Author: fredrec

ci/kokoro/run_bazel_unittests.sh

@@ -33,6 +33,8 @@ 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.

requirements.txt

@@ -5,4 +5,3 @@ scipy
 enum34~=1.1
 mock
 dm-tree~=0.1.1
-tensorflow-compression~=2.8.1