public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-04-13 09:47:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Clear later checkpoints

Browse source
This commit is contained in:
Grant Sanderson 2022-05-01 15:31:42 -04:00
parent 602fbd1a9f
commit 4a8e8e5447
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
manimlib/scene/scene.py
View file

@ -754,6 +754,11 @@ class Scene(object):
if key not in self.checkpoint_states: if key not in self.checkpoint_states:
log.error(f"No checkpoint at {key}") log.error(f"No checkpoint at {key}")
return 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]) self.restore_state(self.checkpoint_states[key])
def clear_checkpoints(self): def clear_checkpoints(self):