saymrwulf/prophet
1
0
Fork 0
mirror of https://github.com/saymrwulf/prophet.git synced 2026-07-30 20:18:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source 
Fix pickling protocol to use highest available.
This commit is contained in:
Sean J. Taylor 2017-03-08 12:37:23 -08:00 committed by GitHub
commit c22294e1c4
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
python/setup.py
View file

@ -31,7 +31,7 @@ class BuildPyCommand(build_py):
model_code = f.read()
sm = StanModel(model_code=model_code)
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):
"""We must run tests on the build directory, not source."""