Use tfp.random.rademacher as tfp.math.random_rademacher is deprecated.

PiperOrigin-RevId: 478812076
google · Oct 4, 2022 · 5338ae3 · 5338ae3
1 parent 5c13e76
commit 5338ae3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/no_u_turn_sampler/nuts.py b/examples/no_u_turn_sampler/nuts.py
@@ -127,7 +127,7 @@ def kernel(target_log_prob_fn,
       # Grow the No-U-Turn Sampler trajectory by choosing a random direction and
       # simulating Hamiltonian dynamics in that direction. This extends either
       # the forward or reverse state.
-      direction = tfp.math.random_rademacher([1], seed=seed_stream())
+      direction = tfp.random.rademacher([1], seed=seed_stream())
       if direction < 0:
         [
             reverse_state,