fix data type problem

sony · May 10, 2018 · 6ca02d1 · 6ca02d1
1 parent cb3b8f0
commit 6ca02d1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/python/test/function/test_embed.py b/python/test/function/test_embed.py
@@ -29,8 +29,8 @@ def test_embed_forward_backward(seed, shape_x, shape_w, ctx, func_name):
     from nbla_test_utils import cap_ignore_region, function_tester
     rng = np.random.RandomState(seed)
     n_class = shape_w[0]
-    x = np.random.randint(0, n_class - 1, shape_x)
-    w = np.random.randn(*shape_w)
+    x = np.random.randint(0, n_class - 1, shape_x).astype(np.int32)
+    w = np.random.randn(*shape_w).astype(np.float32)
     inputs = [x, w]
     function_tester(rng, F.embed, lambda x, w: w[x], inputs,
                     ctx=ctx, func_name=func_name, atol_b=1e-2,

diff --git a/python/test/function/test_logical.py b/python/test/function/test_logical.py
@@ -76,7 +76,7 @@ def test_logical_binary_forward_backward(seed, fname, ctx, func_name):
     func = getattr(F, fname)
     ref_func = getattr(np, fname)
     rng = np.random.RandomState(seed)
-    inputs = [rng.randint(0, 2, size=(2, 3, 4)) for _ in range(2)]
+    inputs = [rng.randint(0, 2, size=(2, 3, 4)).astype(np.float32) for _ in range(2)]
     function_tester(rng, func, ref_func, inputs,
                     ctx=ctx, backward=[False, False], func_name=func_name)