Reformatted

src/jimgw/single_event/runManager.py

@@ -103,7 +103,9 @@ def __init__(self, **kwargs):
 
         local_prior = self.initialize_prior()
         sample_transforms, likelihood_transforms = self.initialize_transforms()
-        local_likelihood = self.initialize_likelihood(local_prior, sample_transforms, likelihood_transforms)
+        local_likelihood = self.initialize_likelihood(
+            local_prior, sample_transforms, likelihood_transforms
+        )
         self.jim = Jim(
             local_likelihood,
             local_prior,
@@ -124,7 +126,12 @@ def load_from_path(self, path: str) -> SingleEventRun:
 
     ### Initialization functions ###
 
-    def initialize_likelihood(self, prior: prior.CombinePrior, sample_transforms: transforms.Transform, likelihood_transforms: transforms.Transform) -> SingleEventLiklihood:
+    def initialize_likelihood(
+        self,
+        prior: prior.CombinePrior,
+        sample_transforms: transforms.Transform,
+        likelihood_transforms: transforms.Transform,
+    ) -> SingleEventLiklihood:
         """
         Since prior contains information about types, naming and ranges of parameters,
         some of the likelihood class require the prior to be initialized, such as the
@@ -473,4 +480,4 @@ def save_summary(self, path: str = "", **kwargs):
             networkSNR = jnp.sum(jnp.array(self.SNRs) ** 2) ** (0.5)
             print("network SNR is", networkSNR)
         sys.stdout.close()
-        sys.stdout=orig_stdout 
+        sys.stdout = orig_stdout