update test_checkpoint score function param name

veugene · Feb 28, 2018 · 54a1507 · 54a1507
1 parent d1e7824
commit 54a1507
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/ignite/handlers/test_checkpoint.py b/tests/ignite/handlers/test_checkpoint.py
@@ -93,7 +93,7 @@ def test_last_k(dirname):
 def test_best_k(dirname):
     scores = iter([1.0, -2., 3.0, -4.0])
 
-    def score_function(engine):
+    def score_function(state):
         return next(scores)
 
     h = ModelCheckpoint(dirname, _PREFIX, create_dir=False,