From 12b5053e11b3d9c9ad4948d8cd549cc9f7d65fe5 Mon Sep 17 00:00:00 2001 From: Sean Morgan Date: Thu, 14 Mar 2019 21:30:11 -0400 Subject: [PATCH] FIX: testutil decorator (#92) * DOC: Add pip install * FIX: un-named argument for new decorator --- tensorflow_addons/layers/python/sparsemax_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_addons/layers/python/sparsemax_test.py b/tensorflow_addons/layers/python/sparsemax_test.py index 3cb375418e..6796982c2f 100644 --- a/tensorflow_addons/layers/python/sparsemax_test.py +++ b/tensorflow_addons/layers/python/sparsemax_test.py @@ -60,7 +60,7 @@ def test_sparsemax_layer_against_numpy(self, dtype=None): z = random.uniform(low=-3, high=3, size=(test_obs, 10)).astype(dtype) test_utils.layer_test( - layer_cls=Sparsemax, + Sparsemax, input_data=z, expected_output=_np_sparsemax(z).astype(dtype))