From 5c0b25f85afaba4b02efb452124e8d08e7cdf351 Mon Sep 17 00:00:00 2001
From: guosheng <guosheng@baidu.com>
Date: Mon, 27 Aug 2018 15:46:13 +0800
Subject: [PATCH] Change random to np.random to be consistent between py2 and
 py3 in Transformer

---
 fluid/neural_machine_translation/transformer/reader.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/fluid/neural_machine_translation/transformer/reader.py b/fluid/neural_machine_translation/transformer/reader.py
index 5c13d89e..4d61affe 100644
--- a/fluid/neural_machine_translation/transformer/reader.py
+++ b/fluid/neural_machine_translation/transformer/reader.py
@@ -1,8 +1,9 @@
 import glob
 import os
-import random
 import tarfile
 
+import numpy as np
+
 
 class SortType(object):
     GLOBAL = 'global'
@@ -203,7 +204,8 @@ class DataReader(object):
         self._token_delimiter = token_delimiter
         self.load_src_trg_ids(end_mark, fpattern, start_mark, tar_fname,
                               unk_mark)
-        self._random = random.Random(x=seed)
+        self._random = np.random
+        self._random.seed(seed)
 
     def load_src_trg_ids(self, end_mark, fpattern, start_mark, tar_fname,
                          unk_mark):
-- 
GitLab