Fixed pickle load encoding from utf-8 to raw binary

tgsmith61591 · Nov 12, 2016 · eb484fd · eb484fd
1 parent bfed8b1
commit eb484fd
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/skutil/h2o/base.py b/skutil/h2o/base.py
@@ -457,7 +457,7 @@ class that is being loaded should be statically referenced at the level of
         m : BaseH2OFunctionWrapper
             The unpickled instance of the model
         """
-        with open(location) as f:
+        with open(location, 'rb') as f:
             m = pickle.load(f)
         return m
 

diff --git a/skutil/h2o/grid_search.py b/skutil/h2o/grid_search.py
@@ -765,7 +765,7 @@ def load(location):
         model : BaseH2OSearchCV
             The unpickled instance of the BaseH2OSearchCV model
         """
-        with open(location) as f:
+        with open(location, 'rb') as f:
             model = pickle.load(f)
 
         if not isinstance(model, BaseH2OSearchCV):

diff --git a/skutil/h2o/pipeline.py b/skutil/h2o/pipeline.py
@@ -480,7 +480,7 @@ def load(location):
         model : H2OPipeline
             The unpickled instance of the H2OPipeline model
         """
-        with open(location) as f:
+        with open(location, 'rb') as f:
             model = pickle.load(f)
 
         if not isinstance(model, H2OPipeline):