Type annotations on scene/vector_space_scene.py

Author: henrikmidtiby

manim/animation/transform.py

@@ -519,7 +519,7 @@ def __init__(
         function: types.MethodType,
         mobject: Mobject,
         run_time: float = DEFAULT_POINTWISE_FUNCTION_RUN_TIME,
-        **kwargs,
+        **kwargs: Any,
     ) -> None:
         super().__init__(mobject.apply_function, function, run_time=run_time, **kwargs)
 

manim/mobject/matrix.py

@@ -410,7 +410,7 @@ def get_mob_matrix(self):
         """
         return self.mob_matrix
 
-    def get_entries(self):
+    def get_entries(self) -> VGroup:
         """Return the individual entries of the matrix.
 
         Returns
@@ -435,7 +435,7 @@ def construct(self):
         """
         return self.elements
 
-    def get_brackets(self):
+    def get_brackets(self) -> VGroup:
         r"""Return the bracket mobjects.
 
         Returns

manim/scene/scene.py

@@ -55,6 +55,8 @@
     from collections.abc import Sequence
     from typing import Callable
 
+    from typing_extensions import Self
+
     from manim.mobject.mobject import _AnimationBuilder
 
 
@@ -813,7 +815,7 @@ def bring_to_back(self, *mobjects: Mobject):
         self.mobjects = list(mobjects) + self.mobjects
         return self
 
-    def clear(self):
+    def clear(self) -> Self:
         """
         Removes all mobjects present in self.mobjects
         and self.foreground_mobjects from the scene.