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

Add minor fix to OutputSplitter.write()

Browse Source
This commit is contained in:
Jonas Triki 2020-12-29 15:00:55 +01:00
parent 6490f5361d
commit 0458f0c1fd
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())
else:
self.file.write(data)
def close(self): def close(self):
self.file.close() self.file.close()