diff options
Diffstat (limited to 'tensorflow/contrib/data/python/ops/resampling.py')
-rw-r--r-- | tensorflow/contrib/data/python/ops/resampling.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/data/python/ops/resampling.py b/tensorflow/contrib/data/python/ops/resampling.py index a182dddd38..b465397437 100644 --- a/tensorflow/contrib/data/python/ops/resampling.py +++ b/tensorflow/contrib/data/python/ops/resampling.py @@ -110,6 +110,7 @@ def rejection_resample(class_func, target_dist, initial_dist=None, seed=None): .filter(lambda _1, p, _2: random_ops.random_uniform([], seed=seed) < p)) return filtered_ds.map(lambda class_value, _, data: (class_value, data)) + return _apply_fn |