refine code

b2de6589 · zenghsh3 · a64200f8 · b2de6589
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/tests/unittests/test_sampling_id_op.py python/paddle/fluid/tests/unittests/test_sampling_id_op.py +1 -1

未找到文件。
--- 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}