refine code

7 years ago · b2de658916
parent a64200f8c1
commit b2de658916
1 changed files with 1 additions and 1 deletions
--- 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}