Update json_dump.py (#13)

A missing dot in ```self.compression``` results in json chunks without compression.
kensho-technologies · Mar 27, 2019 · f5cfd77 · f5cfd77
1 parent 1aef254
commit f5cfd77
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/qwikidata/json_dump.py b/qwikidata/json_dump.py
@@ -88,11 +88,11 @@ def _write_chunk(
             elif out_format == "jsonl":
                 fp.write("\n".join(out_lines))
 
-        if self.compression == "bz2":
+        if self.compression == ".bz2":
             args = ["bzip2", out_fname]
             subprocess.check_output(args)
             out_fname = f"{out_fname}.bz2"
-        elif self.compression == "gz":
+        elif self.compression == ".gz":
             args = ["gzip", out_fname]
             subprocess.check_output(args)
             out_fname = f"{out_fname}.gz"