pylint

tensorflow · Feb 6, 2024 · b674fb1 · b674fb1
1 parent effdad9
commit b674fb1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tensorflow_probability/python/experimental/mcmc/particle_filter_test.py b/tensorflow_probability/python/experimental/mcmc/particle_filter_test.py
@@ -725,7 +725,9 @@ def particle_dynamics(params, _, previous_state):
       broadcasted_params = tf.broadcast_to(reshaped_params,
                                            previous_state.shape)
       reshaped_dist = independent.Independent(
-          normal.Normal(previous_state + params[..., tf.newaxis, tf.newaxis] + 1, 0.1),
+          normal.Normal(
+              previous_state + params[..., tf.newaxis, tf.newaxis] + 1, 0.1
+          ),
           reinterpreted_batch_ndims=1
       )
       return reshaped_dist