diff options
Diffstat (limited to 'lib/parameters.py')
| -rw-r--r-- | lib/parameters.py | 6 |
1 files 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: |