Merge pull request #8 from AkerBP/dev

formatting fix
AkerBP · Aug 27, 2024 · 88b95f2 · 88b95f2
2 parents 0100031 + 79a7011
commit 88b95f2
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/seathru/seathru_model.py b/seathru/seathru_model.py
@@ -498,10 +498,9 @@ def get_loss_dict(self, outputs, batch, metrics_dict=None):
                 raise ValueError(f"Unknown prior_on: {self.config.prior_on}")
 
             # Proposal loss
-            loss_dict[
-                "interlevel_loss"
-            ] = self.config.interlevel_loss_mult * interlevel_loss(
-                outputs["weights_list"], outputs["ray_samples_list"]
+            loss_dict["interlevel_loss"] = (
+                self.config.interlevel_loss_mult
+                * interlevel_loss(outputs["weights_list"], outputs["ray_samples_list"])
             )
         return loss_dict