Merge pull request #22 from NVlabs/develop

Develop
NVlabs · Oct 12, 2022 · dd0a047 · dd0a047
2 parents d71b176 + c944ffa
commit dd0a047
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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]