Skip to content

Commit 9d0a85a

Browse files
fix: Disallow cloudpickle v1.5.0 in 'tensorflow' extra (#915)
* Explicitly disallow cloudpickle v1.5.0 to avoid breaking TensorFlow Probability - This is a temporary solution to unblock development, but should be removed once TensorFlow Probability v0.11 has been released - tensorflow/probability#991 - cloudpipe/cloudpickle#390
1 parent 6a1f996 commit 9d0a85a

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

setup.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,11 @@
11
from setuptools import setup
22

33
extras_require = {
4-
'tensorflow': ['tensorflow~=2.0', 'tensorflow-probability~=0.8'],
4+
'tensorflow': [
5+
'tensorflow~=2.0',
6+
'tensorflow-probability~=0.8',
7+
'cloudpickle!=1.5.0', # TODO: Temp patch until tfp v0.11
8+
],
59
'torch': ['torch~=1.2'],
610
'jax': ['jax~=0.1,>0.1.51', 'jaxlib~=0.1,>0.1.33'],
711
'xmlio': ['uproot'],

0 commit comments

Comments
 (0)