diff --git a/dream/__init__.py b/dream/__init__.py index ec2b805..5c9502a 100644 --- a/dream/__init__.py +++ b/dream/__init__.py @@ -1,4 +1,4 @@ -__version__ = "1.1.0" +__version__ = "1.2.0" from .analysis import * from .datasets import * diff --git a/dream/network.py b/dream/network.py index dccf2ea..9db1a29 100644 --- a/dream/network.py +++ b/dream/network.py @@ -553,7 +553,7 @@ def inference(self, network_input): if self.use_belief_peak_scores and len(peak) > 1: # Try to use the belief map scores peak_sorted_by_score = sorted( - peak, key=lambda x: x[1], reverse=True + peak, key=lambda x: x[2], reverse=True ) if ( peak_sorted_by_score[0][2] - peak_sorted_by_score[1][2]