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

srvasude · edward-bot · commit 5338ae3244b9 · 2022-10-04T09:42:42.000-07:00
PiperOrigin-RevId: 478812076
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,

Original file line number	Diff line number	Diff line change
`@@ -127,7 +127,7 @@ def kernel(target_log_prob_fn,`
`127`	`127`	`# Grow the No-U-Turn Sampler trajectory by choosing a random direction and`
`128`	`128`	`# simulating Hamiltonian dynamics in that direction. This extends either`
`129`	`129`	`# the forward or reverse state.`
`130`		`- direction = tfp.math.random_rademacher([1], seed=seed_stream())`
	`130`	`+ direction = tfp.random.rademacher([1], seed=seed_stream())`
`131`	`131`	`if direction < 0:`
`132`	`132`	`[`
`133`	`133`	`reverse_state,`