From 4e90a77fcdb1a886078561d57fb169d8f552442f Mon Sep 17 00:00:00 2001
From: Grant Sanderson <grant@3blue1brown.com>
Date: Sat, 4 Feb 2023 16:50:12 -0800
Subject: [PATCH] Change type hint on LaggedStart to accept any functions
 outputting animations

---
 manimlib/animation/composition.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/manimlib/animation/composition.py b/manimlib/animation/composition.py
index a1c0ed37..e4ad5656 100644
--- a/manimlib/animation/composition.py
+++ b/manimlib/animation/composition.py
@@ -165,7 +165,7 @@ class LaggedStart(AnimationGroup):
 class LaggedStartMap(LaggedStart):
     def __init__(
         self,
-        AnimationClass: type,
+        anim_func: Callable[[Mobject], Animation],
         group: Mobject,
         arg_creator: Callable[[Mobject], tuple] | None = None,
         run_time: float = 2.0,
@@ -175,7 +175,7 @@ class LaggedStartMap(LaggedStart):
         anim_kwargs = dict(kwargs)
         anim_kwargs.pop("lag_ratio", None)
         super().__init__(
-            *(AnimationClass(submob, **anim_kwargs) for submob in group),
+            *(anim_func(submob, **anim_kwargs) for submob in group),
             run_time=run_time,
             lag_ratio=lag_ratio,
         )