Skip to content

Commit c624cad

Browse files
jburnimemilyfertig
authored andcommitted
Always use cloudpickle instead of cloudpickle_fast, even in Python 3.8+.
PiperOrigin-RevId: 311330869
1 parent 3ec36f2 commit c624cad

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow_probability/python/layers/distribution_layer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
import pickle
2626

2727
# Dependency imports
28-
from cloudpickle import CloudPickler
28+
from cloudpickle.cloudpickle import CloudPickler
2929
import numpy as np
3030
import six
3131
import tensorflow.compat.v2 as tf

0 commit comments

Comments
 (0)