From b2de658916819a168898fc36fd8f463537f0612c Mon Sep 17 00:00:00 2001 From: zenghsh3 Date: Fri, 31 Aug 2018 17:18:31 +0800 Subject: [PATCH] refine code --- python/paddle/fluid/tests/unittests/test_sampling_id_op.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/tests/unittests/test_sampling_id_op.py b/python/paddle/fluid/tests/unittests/test_sampling_id_op.py index 6e0c26943aa..981958ee089 100644 --- a/python/paddle/fluid/tests/unittests/test_sampling_id_op.py +++ b/python/paddle/fluid/tests/unittests/test_sampling_id_op.py @@ -27,7 +27,7 @@ class TestSamplingIdOp(OpTest): self.init_kernel_type() self.X = np.random.random((100, 10)).astype('float32') self.inputs = {"X": self.X} - self.Y = np.random.random(100).astype('float32') + self.Y = np.random.random(100).astype('int64') self.outputs = {'Out': self.Y} self.attrs = {'max': 1.0, 'min': 0.0, 'seed': 1} -- GitLab