Update random.py

Ran pre-commit.
aesara-devs · Dec 12, 2022 · b77cf5d · b77cf5d
1 parent ed2b8fc
commit b77cf5d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/aesara/link/jax/dispatch/random.py b/aesara/link/jax/dispatch/random.py
@@ -194,7 +194,9 @@ def sample_fn(rng, size, dtype, *parameters):
         rng_key = rng["jax_state"]
         rng_key, sampling_key = jax.random.split(rng_key, 2)
         minval, maxval = parameters
-        sample = jax_op(sampling_key, shape=size, dtype=dtype, minval=minval, maxval=maxval)
+        sample = jax_op(
+            sampling_key, shape=size, dtype=dtype, minval=minval, maxval=maxval
+        )
         rng["jax_state"] = rng_key
         return (rng, sample)