From 416bbe026cc0f4aae25a5b2062459c319179886f Mon Sep 17 00:00:00 2001 From: Valentin Date: Wed, 6 Dec 2017 21:36:06 +0100 Subject: [PATCH] add lines for pep8 --- keras_contrib/layers/advanced_activations.py | 5 ++++- tests/keras_contrib/layers/test_advanced_activations.py | 1 + 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/keras_contrib/layers/advanced_activations.py b/keras_contrib/layers/advanced_activations.py index 894166b..7bf349e 100644 --- a/keras_contrib/layers/advanced_activations.py +++ b/keras_contrib/layers/advanced_activations.py @@ -235,6 +235,9 @@ class SReLU(Layer): base_config = super(SReLU, self).get_config() return dict(list(base_config.items()) + list(config.items())) +get_custom_objects().update({'SReLU': SReLU}) + + class Swish(Layer): """ Swish (Ramachandranet al., 2017) @@ -279,4 +282,4 @@ class Swish(Layer): base_config = super(Swish, self).get_config() return dict(list(base_config.items()) + list(config.items())) -get_custom_objects().update({'SReLU': SReLU, 'Swish': Swish}) +get_custom_objects().update({'Swish': Swish}) diff --git a/tests/keras_contrib/layers/test_advanced_activations.py b/tests/keras_contrib/layers/test_advanced_activations.py index 57e3d4a..8c71426 100644 --- a/tests/keras_contrib/layers/test_advanced_activations.py +++ b/tests/keras_contrib/layers/test_advanced_activations.py @@ -26,6 +26,7 @@ def test_srelu_share(): layer_test(advanced_activations.SReLU, kwargs={'shared_axes': 1}, input_shape=(2, 3, 4)) + @keras_test def test_swish_constant(): layer_test(advanced_activations.Swish, kwargs={'beta': 1.0, 'trainable': False},