From 4a8e8e5447c5496d0db5cd7934a71badfcf3ce8d Mon Sep 17 00:00:00 2001
From: Grant Sanderson <grant@3blue1brown.com>
Date: Sun, 1 May 2022 15:31:42 -0400
Subject: [PATCH] Clear later checkpoints

---
 manimlib/scene/scene.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/manimlib/scene/scene.py b/manimlib/scene/scene.py
index 71f40e83..a1c7af76 100644
--- a/manimlib/scene/scene.py
+++ b/manimlib/scene/scene.py
@@ -754,6 +754,11 @@ class Scene(object):
         if key not in self.checkpoint_states:
             log.error(f"No checkpoint at {key}")
             return
+        all_keys = list(self.checkpoint_states.keys())
+        index = all_keys.index(key)
+        for later_key in all_keys[index + 1:]:
+            self.checkpoint_states.pop(later_key)
+
         self.restore_state(self.checkpoint_states[key])
 
     def clear_checkpoints(self):