Merge pull request #5147 from takagi/fix-contrastive-test

Trivial fix for parameterized test case of F.contrastive
chainer · Jul 26, 2018 · c09441d · c09441d
2 parents d321a32 + ce5d0f6
commit c09441d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/chainer_tests/functions_tests/loss_tests/test_contrastive.py b/tests/chainer_tests/functions_tests/loss_tests/test_contrastive.py
@@ -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):