From da127f2a8f5262efbdb5d4a8afcb8cdc58b8dbfa Mon Sep 17 00:00:00 2001 From: Jonas Eschle Date: Wed, 16 Oct 2024 11:13:22 -0400 Subject: [PATCH] chore: reformat file --- tf_pwa/model/model.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tf_pwa/model/model.py b/tf_pwa/model/model.py index fe69c06..c9cf04d 100644 --- a/tf_pwa/model/model.py +++ b/tf_pwa/model/model.py @@ -327,7 +327,9 @@ def nll(self, data, mcdata): amp_s2 = self.signal(data) * weight amp_s2 = self.sum_resolution(amp_s2) weight = tf.reduce_sum(rw, axis=-1) - dom_weight = tf.where(weight == 0, tf.constant(1.0, dtype=weight.dtype), weight) + dom_weight = tf.where( + weight == 0, tf.constant(1.0, dtype=weight.dtype), weight + ) ln_data = clip_log(amp_s2 / dom_weight) mc_weight = mcdata.get("weight", tf.ones((data_shape(mcdata),))) int_mc = tf.reduce_sum(