Add support for TensorFlow 2.8

Author: drasmuss

.nengobones.yml

@@ -59,6 +59,8 @@ docs_conf_py:
   doctest_setup:
     - import numpy as np
     - import tensorflow as tf
+  sphinx_options:
+    autodoc_inherit_docstrings: False
 
 travis_yml:
   python: 3.8

CHANGES.rst

@@ -22,10 +22,19 @@ Release history
 0.4.1 (unreleased)
 ==================
 
+*Compatible with TensorFlow 2.1 - 2.8*
+
+**Added**
+
+- Added support for TensorFlow 2.8. (`#46`_)
+
+.. _#46: https://github.com/nengo/keras-lmu/pull/46
 
 0.4.0 (August 16, 2021)
 =======================
 
+*Compatible with TensorFlow 2.1 - 2.7*
+
 **Added**
 
 - Setting ``kernel_initializer=None`` now removes the dense input kernel. (`#40`_)

docs/conf.py

@@ -85,6 +85,7 @@
 default_role = "py:obj"
 pygments_style = "sphinx"
 user_agent = "keras_lmu"
+autodoc_inherit_docstrings = False
 
 project = "KerasLMU"
 authors = "Applied Brain Research"

keras_lmu/layers.py

@@ -11,8 +11,15 @@
 else:
     from keras.layers.recurrent import DropoutRNNCellMixin
 
+if version.parse(tf.__version__) < version.parse(
+    "2.8.0rc0"
+):  # pylint: disable=ungrouped-imports
+    from tensorflow.keras.layers import Layer as BaseRandomLayer
+else:
+    from keras.engine.base_layer import BaseRandomLayer
+
 
-class LMUCell(DropoutRNNCellMixin, tf.keras.layers.Layer):
+class LMUCell(DropoutRNNCellMixin, BaseRandomLayer):
     """
     Implementation of LMU cell (to be used within Keras RNN wrapper).
 

keras_lmu/tests/conftest.py

@@ -0,0 +1,9 @@
+# pylint: disable=missing-docstring
+
+import tensorflow as tf
+from packaging import version
+
+
+def pytest_configure(config):
+    if version.parse(tf.__version__) >= version.parse("2.7.0"):
+        tf.debugging.disable_traceback_filtering()