From 7c57501ad19412376399870666cf560cd123cb73 Mon Sep 17 00:00:00 2001 From: Vincent Roseberry Date: Tue, 19 May 2020 22:07:19 +0000 Subject: [PATCH] Add TensorFlow addons (TFA) Note: TFA functionnality relying on Custom Op won't work. See: https://github.com/tensorflow/addons/issues/987. BUG=145555176 --- Dockerfile | 5 +++++ tests/test_tensorflow_addons.py | 14 ++++++++++++++ 2 files changed, 19 insertions(+) create mode 100644 tests/test_tensorflow_addons.py diff --git a/Dockerfile b/Dockerfile index a04401f7..69c28dba 100644 --- a/Dockerfile +++ b/Dockerfile @@ -65,6 +65,11 @@ RUN pip install /tmp/tensorflow_gcs_config/tensorflow*.whl && \ rm -rf /tmp/tensorflow_gcs_config && \ /tmp/clean-layer.sh +# Install TensorFlow addons (TFA). +# TFA functionnality relying on Custom Op won't work. See: https://github.com/tensorflow/addons/issues/987. +RUN pip install tensorflow-addons==0.10.0 && \ + /tmp/clean-layer.sh + RUN apt-get install -y libfreetype6-dev && \ apt-get install -y libglib2.0-0 libxext6 libsm6 libxrender1 libfontconfig1 --fix-missing && \ pip install gensim && \ diff --git a/tests/test_tensorflow_addons.py b/tests/test_tensorflow_addons.py new file mode 100644 index 00000000..51a09489 --- /dev/null +++ b/tests/test_tensorflow_addons.py @@ -0,0 +1,14 @@ +import unittest + +import tensorflow as tf +import tensorflow_addons as tfa + + +class TestTensorflowAddons(unittest.TestCase): + def test_tfa_image(self): + img_raw = tf.io.read_file('/input/tests/data/dot.png') + img = tf.io.decode_image(img_raw) + img = tf.image.convert_image_dtype(img, tf.float32) + mean = tfa.image.mean_filter2d(img, filter_shape=1) + + self.assertEqual(1, len(mean)) \ No newline at end of file