fix bugs

go/master/client.go

@@ -113,7 +113,7 @@ func (c *Client) monitorMaster(addr Addresser) {
 //
 // SetDataset can be call multiple times from different nodes. But
 // only the first call will be honored.
-func (c *Client) SetDataset(globPaths ...string) error {
+func (c *Client) SetDataset(globPaths []string) error {
 	return c.conn.Call("Service.SetDataset", globPaths, nil)
 }
 

python/paddle/v2/reader/creator.py

@@ -93,13 +93,17 @@ def recordio(paths, addr="", buf_size=100):
     if len(os.environ["KUBERNETES_SERVICE_HOST"]) == 0:
         return recordio_local(path)
 
-    c = cloud(addr, buf_size)
+    def reader():
-    c.set_dataset(paths)
+        c = cloud(addr, buf_size)
+        c.set_dataset(paths)
+
+        while True:
+            r = client.next_record()
+            if r is None:
+                break
+            yield r
 
-    while True:
+        c.close()
-        r = client.next_record()
+
-        if r is None:
+    return reader
-            break
-        yield r
 
-    c.close()