Revert "Revert "Always use cloudpickle instead of cloudpickle_fast, e…

…ven in Python 3.8+."" This reverts commit fa66f19.
tensorflow · Jul 6, 2020 · c087524 · c087524
1 parent fa66f19
commit c087524
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_probability/python/layers/distribution_layer.py b/tensorflow_probability/python/layers/distribution_layer.py
@@ -25,7 +25,7 @@
 import pickle
 
 # Dependency imports
-from cloudpickle import CloudPickler
+from cloudpickle.cloudpickle import CloudPickler
 import numpy as np
 import six
 import tensorflow.compat.v2 as tf