Add TensorFlow addons (TFA)

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 && \

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))