From 14b59f86359d83af700e8d2e91168bb83a148c27 Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Tue, 16 Jan 2024 18:18:36 +0100 Subject: build_dtree: add another assertion for good measure --- lib/parameters.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/lib/parameters.py b/lib/parameters.py index 58af96f..a32eb9f 100644 --- a/lib/parameters.py +++ b/lib/parameters.py @@ -1303,10 +1303,8 @@ class ModelAttribute: else: children.extend((np.array(child_data) - np.mean(child_data)) ** 2) - if np.any(np.isnan(children)): - loss.append(np.inf) - else: - loss.append(np.sum(children)) + assert not np.any(np.isnan(children)) + loss.append(np.sum(children)) if np.all(np.isinf(loss)): if ffs_feasible: -- cgit v1.2.3