public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-05 16:49:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add type hints for affects_mobject_list

Browse source
This commit is contained in:
Grant Sanderson 2024-03-02 16:49:35 -05:00
parent 7edc4b64ad
commit ffbe5c8114
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
manimlib/scene/scene.py
View file

@ -44,9 +44,11 @@ from manimlib.utils.iterables import batch_by_property
from typing import TYPE_CHECKING from typing import TYPE_CHECKING
if TYPE_CHECKING: if TYPE_CHECKING:
from typing import Callable, Iterable from typing import Callable, Iterable, TypeVar
from manimlib.typing import Vect3 from manimlib.typing import Vect3
T = TypeVar('T')
from PIL.Image import Image from PIL.Image import Image
from manimlib.animation.animation import Animation from manimlib.animation.animation import Animation
@ -399,7 +401,8 @@ class Scene(object):
for batch, key in batches for batch, key in batches
] ]
def affects_mobject_list(func: Callable): @staticmethod
def affects_mobject_list(func: Callable[..., T]) -> Callable[..., T]:
@wraps(func) @wraps(func)
def wrapper(self, *args, **kwargs): def wrapper(self, *args, **kwargs):
func(self, *args, **kwargs) func(self, *args, **kwargs)