diff --git a/Dockerfile b/Dockerfile index 3b60d900..c56fa1fb 100644 --- a/Dockerfile +++ b/Dockerfile @@ -442,6 +442,7 @@ RUN pip install flashtext && \ # papermill can replace nbconvert for executing notebooks pip install papermill && \ pip install cloud-tpu-client && \ + pip install tensorflow-cloud && \ pip install tensorflow-datasets && \ pip install pydub && \ pip install pydegensac && \ diff --git a/tests/test_tensorflow_cloud.py b/tests/test_tensorflow_cloud.py new file mode 100644 index 00000000..2875e121 --- /dev/null +++ b/tests/test_tensorflow_cloud.py @@ -0,0 +1,8 @@ +import unittest + +import tensorflow_cloud as tfc + + +class TestTensorflowCloud(unittest.TestCase): + def test_remote(self): + self.assertFalse(tfc.remote())