From 0c7764b0100e2846f0fb8a32f4683b3832df7218 Mon Sep 17 00:00:00 2001
From: Sven Mika <sven@anyscale.io>
Date: Tue, 16 Jun 2020 08:51:20 +0200
Subject: [PATCH] Issue 8919 checkpoint at end ignored (#8933)

---
 rllib/tests/test_checkpoint_restore.py | 3 +++
 rllib/train.py                         | 1 +
 2 files changed, 4 insertions(+)

diff --git a/rllib/tests/test_checkpoint_restore.py b/rllib/tests/test_checkpoint_restore.py
index 994a07b84..276a28d25 100644
--- a/rllib/tests/test_checkpoint_restore.py
+++ b/rllib/tests/test_checkpoint_restore.py
@@ -124,6 +124,9 @@ def ckpt_restore_test(alg_name, tfe=False):
                 if abs(a1 - a2) > .1:
                     raise AssertionError("algo={} [a1={} a2={}]".format(
                         alg_name, a1, a2))
+            # Stop both Trainers.
+            alg1.stop()
+            alg2.stop()
 
 
 class TestCheckpointRestore(unittest.TestCase):
diff --git a/rllib/train.py b/rllib/train.py
index 0932e08b2..54f91a210 100755
--- a/rllib/train.py
+++ b/rllib/train.py
@@ -149,6 +149,7 @@ def run(args, parser):
             args.experiment_name: {  # i.e. log to ~/ray_results/default
                 "run": args.run,
                 "checkpoint_freq": args.checkpoint_freq,
+                "checkpoint_at_end": args.checkpoint_at_end,
                 "keep_checkpoints_num": args.keep_checkpoints_num,
                 "checkpoint_score_attr": args.checkpoint_score_attr,
                 "local_dir": args.local_dir,