Merge pull request #44 from martindurant/object_types

Specify object types
dask · Dec 13, 2016 · ea1680d · ea1680d
2 parents 9179be6 + dfd025f
commit ea1680d
Show file tree

Hide file tree

Showing 2 changed files with 129 additions and 118 deletions.
diff --git a/fastparquet/converted_types.py b/fastparquet/converted_types.py
@@ -28,13 +28,17 @@
 
 try:
     from bson import BSON
-    unbson = lambda s: BSON(s).decode()
+    unbson = bson.BSON.decode
+    tobson = bson.BSON.encode
 except ImportError:
     try:
         import bson
         unbson = bson.loads
+        tobson = bson.dumps
     except:
-        def BSON(x):
+        def unbson(x):
+            raise ImportError("BSON not found")
+        def tobson(x):
             raise ImportError("BSON not found")
 
 DAYS_TO_MILLIS = 86400000000000