diff --git a/python/paddle/v2/dataset/mq2007.py b/python/paddle/v2/dataset/mq2007.py index d8c9918d1401232d7a59d0dfb5877f2010f51ea3..fd71b341662ca6f540ce44a86348e782561a97d7 100644 --- a/python/paddle/v2/dataset/mq2007.py +++ b/python/paddle/v2/dataset/mq2007.py @@ -41,7 +41,9 @@ def __initialize_meta_info__(): """ fn = fetch() rar = rarfile.RarFile(fn) - return rar + dirpath = os.path.dirname(fn) + rar.extractall(path=dirpath) + return dirpath class Query(object): @@ -271,7 +273,7 @@ def load_from_text(filepath, shuffle=True, fill_missing=-1): querylists = [] querylist = None fn = __initialize_meta_info__() - with fn.open(os.path.join(fn, filepath)) as f: + with open(os.path.join(fn, filepath)) as f: for line in f: query = Query() query = query._parse_(line)