Black formatting

example/GW150914.py

@@ -83,7 +83,13 @@
         sin_dec_prior,
     ]
 )
-likelihood = TransientLikelihoodFD([H1, L1], waveform=RippleIMRPhenomD(), trigger_time=gps, duration=4, post_trigger_duration=2)
+likelihood = TransientLikelihoodFD(
+    [H1, L1],
+    waveform=RippleIMRPhenomD(),
+    trigger_time=gps,
+    duration=4,
+    post_trigger_duration=2,
+)
 
 
 mass_matrix = jnp.eye(11)

src/jimgw/prior.py

@@ -193,7 +193,7 @@ def sample(self, rng_key: jax.random.PRNGKey, n_samples: int) -> dict:
 
     def log_prob(self, x: dict) -> Float:
         variable = x[self.naming[0]]
-        return jnp.log(jnp.exp(-variable)/(1 + jnp.exp(-variable))**2) 
+        return jnp.log(jnp.exp(-variable) / (1 + jnp.exp(-variable)) ** 2)
 
 
 class Sphere(Prior):

test/test_prior.py

@@ -1 +1 @@
-from jimgw.prior import Uniform, UUniform, Composite
+from jimgw.prior import Uniform, Unconstrained_Uniform, Composite