Add minor fix to OutputSplitter.write()

2020-12-29 15:00:55 +01:00 · 2020-12-29 15:00:55 +01:00 · 0458f0c1fd
commit 0458f0c1fd
parent 6490f5361d
1 changed files with 4 additions and 1 deletions
--- a/wikiextractor/WikiExtractor.py
+++ b/wikiextractor/WikiExtractor.py
@ -169,7 +169,10 @@ class OutputSplitter():

    def write(self, 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):
        self.file.close()