diff options
Diffstat (limited to 's2-extract-papers.py')
| -rw-r--r-- | s2-extract-papers.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/s2-extract-papers.py b/s2-extract-papers.py index 178b9009..e84ffd0f 100644 --- a/s2-extract-papers.py +++ b/s2-extract-papers.py @@ -41,7 +41,7 @@ def load_id_lookup(fn): return lookup def paper_path(paper_id): - return '{}/{}/{}'.format(DATA_DIR, paper_id[0:3], paper_id) + return '{}/{}/{}'.format(DATA_DIR, paper_id[0:2], paper_id) def write_paper(paper_id, data): dir = paper_path(paper_id) @@ -49,7 +49,7 @@ def write_paper(paper_id, data): if os.path.exists(fn): return os.makedirs(dir, exist_ok=True) - with open(fn, 'wb') as f: + with open(fn, 'w') as f: f.write(data) if __name__ == '__main__': |