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

Fix pickling protocol to use highest available.

Browse source 
This fixes unpickling errors in Python 2.7 as it defaults to the text
protocol.
This commit is contained in:
Rolando Espinoza 2017-03-08 16:34:08 -03:00
parent b7a201086d
commit d446bb1c19
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."""