Py3k fix.

alayers2 · Jul 20, 2016 · 9eee239 · 9eee239
1 parent c8629a6
commit 9eee239
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/matgendb/creator.py b/matgendb/creator.py
@@ -254,7 +254,8 @@ def _insert_doc(self, d):
                         if "dos" in calc:
                             dos = json.dumps(calc["dos"], cls=MontyEncoder)
                             if self.compress_dos:
-                                dos = zlib.compress(dos, self.compress_dos)
+                                dos = zlib.compress(bytes(dos, 'utf-8'),
+                                                    self.compress_dos)
                                 calc["dos_compression"] = "zlib"
                             fs = gridfs.GridFS(db, "dos_fs")
                             dosid = fs.put(dos)

diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,6 @@
-pymatgen==4.0.0
+pymatgen==4.1.1
 monty==0.8.6
-pymongo==3.2.2
+pymongo==3.3.0
 prettytable==0.7.2
 mongomock==3.0.0
 smoqe==0.1.3