From 5d1458ed335f08969d23c5f1321d9ffeec1ef9b9 Mon Sep 17 00:00:00 2001
From: xuyuon <116078673+xuyuon@users.noreply.github.com>
Date: Fri, 30 Aug 2024 10:06:59 +0800
Subject: [PATCH] Reformatted

---
 src/jimgw/single_event/runManager.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/src/jimgw/single_event/runManager.py b/src/jimgw/single_event/runManager.py
index e4d95e08..2fa00cd9 100644
--- a/src/jimgw/single_event/runManager.py
+++ b/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