NLP/wikiextractor
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #241 from JonasTriki/fix-compressed-file-write

Browse Source 
Add minor fix to OutputSplitter.write()
This commit is contained in:
Giuseppe Attardi 2021-03-18 01:14:59 +01:00 committed by GitHub
commit 881f3e4252
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
wikiextractor/WikiExtractor.py
View File

@ -169,7 +169,10 @@ class OutputSplitter():
def write(self, data): def write(self, data):
self.reserve(len(data)) self.reserve(len(data))
self.file.write(data) if self.compress:
self.file.write(data.encode('utf-8'))
else:
self.file.write(data)
def close(self): def close(self):
self.file.close() self.file.close()