Merge pull request #24 from sile/fix-wrong-terminate-condition

Allow the case where `end_step == self.study_step`
optuna · Aug 21, 2020 · 3d06f06 · 3d06f06
2 parents e66ee3e + 7aa3ca6
commit 3d06f06
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/runner.rs b/src/runner.rs
@@ -288,7 +288,7 @@ impl StudyRunner {
         self.pb.inc(elapsed_steps);
         let end_step = self.pb.position();
 
-        if end_step < self.study_steps {
+        if end_step <= self.study_steps {
             let ((), tell_elapsed) =
                 ElapsedSeconds::try_time(|| track!(self.solver.tell(evaluated_trial.clone())))?;