Home Explore Help
Sign In
radu
/
Facebook-Prophet
Watch 1
Star 0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Browse Source

Merge pull request #98 from rolando-contrib/fix-pickle-protocol

Fix pickling protocol to use highest available.
Sean J. Taylor 8 years ago
parent
fcbd957bcc d446bb1c19
commit
c22294e1c4
1 changed files with 1 additions and 1 deletions
Unified View Show Diff Stats
  1. 1 1
      python/setup.py

+ 1 - 1
python/setup.py
View File 

@@ -31,7 +31,7 @@ class BuildPyCommand(build_py):
 
                 model_code = f.read()
 
                 model_code = f.read()
 
             sm = StanModel(model_code=model_code)
 
             sm = StanModel(model_code=model_code)
 
             with open(os.path.join(target_dir, '{}_growth.pkl'.format(model_type)), 'wb') as f:
 
             with open(os.path.join(target_dir, '{}_growth.pkl'.format(model_type)), 'wb') as f:
 
-                pickle.dump(sm, f)
 
+                pickle.dump(sm, f, protocol=pickle.HIGHEST_PROTOCOL)
 
 
 
 class TestCommand(test_command):
 
 class TestCommand(test_command):
 
     """We must run tests on the build directory, not source."""
 
     """We must run tests on the build directory, not source."""