From 13a1117be52eddad88b5ec86044d7e96a6c8edad Mon Sep 17 00:00:00 2001
From: swijaya <swijaya@google.com>
Date: Thu, 21 Mar 2024 15:33:03 -0700
Subject: [PATCH] Fix unbalanced backticks in docstring

PiperOrigin-RevId: 617982396
---
 .../python/experimental/mcmc/windowed_sampling.py         | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/tensorflow_probability/python/experimental/mcmc/windowed_sampling.py b/tensorflow_probability/python/experimental/mcmc/windowed_sampling.py
index ab30e69471..643b409864 100644
--- a/tensorflow_probability/python/experimental/mcmc/windowed_sampling.py
+++ b/tensorflow_probability/python/experimental/mcmc/windowed_sampling.py
@@ -642,9 +642,11 @@ def windowed_adaptive_nuts(n_draws,
       structure should broadcast with `current_state`. For example, if the
       initial state is
       ```
-      {'a': tf.zeros(n_chains),
-       'b': tf.zeros([n_chains, n_features])}
-       ```
+      {
+       'a': tf.zeros(n_chains),
+       'b': tf.zeros([n_chains, n_features]),
+      }
+      ```
       then any of `1.`, `{'a': 1., 'b': 1.}`, or
       `{'a': tf.ones(n_chains), 'b': tf.ones([n_chains, n_features])}` will
       work. Defaults to the dimension of the log density to the 0.25 power.