Add spaces around '/'.

PiperOrigin-RevId: 439774759
google-deepmind · Apr 6, 2022 · 60ce578 · 60ce578
1 parent f4e7298
commit 60ce578
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/distrax/_src/distributions/log_stddev_normal.py b/distrax/_src/distributions/log_stddev_normal.py
@@ -91,7 +91,7 @@ def _kl_logstddevnormal_logstddevnormal(p, q, *unused_args, **unused_kwargs):
   # (s_a^2 + (u_a - u_b)^2)/(2*s_b^2) + log(s_b) - log(s_a) - 1/2
   p_variance = jnp.square(p.scale)
   q_variance = jnp.square(q.scale)
-  return ((p_variance + jnp.square(p.loc - q.loc))/(2.0 * q_variance) +
+  return ((p_variance + jnp.square(p.loc - q.loc)) / (2.0 * q_variance) +
           q.log_scale - p.log_scale - 0.5)