Skip to content

Commit 12b5053

Browse files
seanpmorganfacaiy
authored andcommitted
FIX: testutil decorator (#92)
* DOC: Add pip install * FIX: un-named argument for new decorator
1 parent f36d04f commit 12b5053

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow_addons/layers/python/sparsemax_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def test_sparsemax_layer_against_numpy(self, dtype=None):
6060
z = random.uniform(low=-3, high=3, size=(test_obs, 10)).astype(dtype)
6161

6262
test_utils.layer_test(
63-
layer_cls=Sparsemax,
63+
Sparsemax,
6464
input_data=z,
6565
expected_output=_np_sparsemax(z).astype(dtype))
6666

0 commit comments

Comments
 (0)