Skip to content

Commit

Permalink
Fix parameterized test case
Browse files Browse the repository at this point in the history
  • Loading branch information
takagi committed Jul 25, 2018
1 parent c851f59 commit ce5d0f6
Showing 1 changed file with 1 addition and 1 deletion.
Expand Up @@ -15,7 +15,7 @@
@testing.parameterize(
*testing.product({
'batchsize': [5, 10], 'input_dim': [2, 3], 'margin': [1, 2],
'reduce': ['mean', 'no'], 'label_dtype': [numpy.int32, numpy.int32]
'reduce': ['mean', 'no'], 'label_dtype': [numpy.int32, numpy.int64]
})
)
class TestContrastive(unittest.TestCase):
Expand Down

0 comments on commit ce5d0f6

Please sign in to comment.