Merge pull request #153 from jonas-eschle/patch-1

dtype aligns in nll
jiangyi15 · Oct 16, 2024 · dc0326f · dc0326f
2 parents 07d25ce + da127f2
commit dc0326f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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, 1.0, 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(