fixing doc example

arthurpaulino · Apr 18, 2019 · fbcb742 · fbcb742
1 parent c7d9560
commit fbcb742
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/miraiml/__init__.py b/miraiml/__init__.py
@@ -10,7 +10,7 @@
 >>> from miraiml import HyperSearchSpace, Config, Engine
 """
 
-__version__ = '2.0.1.4'
+__version__ = '2.0.1.5'
 
 from .main import HyperSearchSpace, Config, Engine
 

diff --git a/miraiml/main.py b/miraiml/main.py
@@ -252,9 +252,8 @@ class Engine:
         from miraiml import Engine
 
         def on_improvement(status):
-            best_id = status['best_id']
-            scores = status['scores']
-            print('Best score:', scores[best_id])
+            # prints the best score everytime an improvement is found
+            print('Score:', status['score'])
 
         engine = Engine(config, on_improvement=on_improvement)
     """