diff --git a/manimlib/for_3b1b_videos/pi_creature_scene.py b/manimlib/for_3b1b_videos/pi_creature_scene.py index 3c702efc..bb790691 100644 --- a/manimlib/for_3b1b_videos/pi_creature_scene.py +++ b/manimlib/for_3b1b_videos/pi_creature_scene.py @@ -211,21 +211,24 @@ class PiCreatureScene(Scene): ]) return self - def wait(self, time=1, blink=True): + def wait(self, time=1, blink=True, **kwargs): + if "stop_condition" in kwargs: + self.non_blink_wait(time, **kwargs) + return while time >= 1: time_to_blink = self.total_wait_time % self.seconds_to_blink == 0 if blink and self.any_pi_creatures_on_screen() and time_to_blink: self.blink() else: - self.non_blink_wait() + self.non_blink_wait(**kwargs) time -= 1 self.total_wait_time += 1 if time > 0: - self.non_blink_wait(time) + self.non_blink_wait(time, **kwargs) return self - def non_blink_wait(self, time=1): - Scene.wait(self, time) + def non_blink_wait(self, time=1, **kwargs): + Scene.wait(self, time, **kwargs) return self def change_mode(self, mode):