poisson updated

aertslab · Nov 18, 2024 · 9610701 · 9610701
1 parent 9091d0b
commit 9610701
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/crested/tl/losses/_poisson.py b/src/crested/tl/losses/_poisson.py
@@ -1,6 +1,7 @@
 import keras
 import keras.ops as ops
 
+
 @keras.saving.register_keras_serializable(package="Losses")
 class PoissonLoss(keras.losses.Loss):
     """
@@ -19,6 +20,7 @@ class PoissonLoss(keras.losses.Loss):
     reduction : str
         Type of reduction to apply to the loss. Default: "sum_over_batch_size".
     """
+
     def __init__(
         self,
         log_transform: bool = True,
@@ -68,4 +70,4 @@ def get_config(self):
             "log_transform": self.log_transform,
             "eps": self.eps
         })
-        return config
+        return config