fix iter bug in dump_entities_to_json (#36)

kensho-technologies · Nov 9, 2022 · 04a1867 · 04a1867
1 parent e77f105
commit 04a1867
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/qwikidata/__init__.py b/qwikidata/__init__.py
@@ -2,4 +2,4 @@
 """Metadata for this package."""
 
 __package_name__ = "qwikidata"
-__version__ = "0.4.1"
+__version__ = "0.4.2"
diff --git a/qwikidata/utils.py b/qwikidata/utils.py
@@ -27,12 +27,12 @@ def dump_entities_to_json(entities: Iterable[WikidataEntity], out_fname: str) ->
     """
     with open(out_fname, "w") as fp:
         fp.write("[")
-
-        ent = next(entities, None)
+        ent_iter = iter(entities)
+        ent = next(ent_iter, None)
         while ent:
             ent_str = json.dumps(ent._entity_dict)
             fp.write("\n{}".format(ent_str))
-            ent = next(entities, None)
+            ent = next(ent_iter, None)
             if ent: fp.write(",")
 
         fp.write("\n]")