提交 4874810b 编写于 作者: G gongweibao

fix bugs

上级 183a5d44
...@@ -113,7 +113,7 @@ func (c *Client) monitorMaster(addr Addresser) { ...@@ -113,7 +113,7 @@ func (c *Client) monitorMaster(addr Addresser) {
// //
// SetDataset can be call multiple times from different nodes. But // SetDataset can be call multiple times from different nodes. But
// only the first call will be honored. // 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) return c.conn.Call("Service.SetDataset", globPaths, nil)
} }
......
...@@ -93,6 +93,7 @@ def recordio(paths, addr="", buf_size=100): ...@@ -93,6 +93,7 @@ def recordio(paths, addr="", buf_size=100):
if len(os.environ["KUBERNETES_SERVICE_HOST"]) == 0: if len(os.environ["KUBERNETES_SERVICE_HOST"]) == 0:
return recordio_local(path) return recordio_local(path)
def reader():
c = cloud(addr, buf_size) c = cloud(addr, buf_size)
c.set_dataset(paths) c.set_dataset(paths)
...@@ -103,3 +104,6 @@ def recordio(paths, addr="", buf_size=100): ...@@ -103,3 +104,6 @@ def recordio(paths, addr="", buf_size=100):
yield r yield r
c.close() c.close()
return reader
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册