Type annotations for manim/scene/*.py

manim/animation/transform.py

@@ -49,6 +49,8 @@
 from ..utils.rate_functions import smooth, squish_rate_func
 
 if TYPE_CHECKING:
+    from typing import Any
+
     from ..scene.scene import Scene
 
 
@@ -516,7 +518,7 @@ def construct(self):
 
     def __init__(
         self,
-        function: types.MethodType,
+        function: Callable,
         mobject: Mobject,
         run_time: float = DEFAULT_POINTWISE_FUNCTION_RUN_TIME,
         **kwargs,
@@ -615,7 +617,9 @@ def __init__(self, mobject: Mobject, **kwargs) -> None:
 
 
 class ApplyFunction(Transform):
-    def __init__(self, function: types.MethodType, mobject: Mobject, **kwargs) -> None:
+    def __init__(
+        self, function: Callable[[Any], Any], mobject: Mobject, **kwargs: Any
+    ) -> None:
         self.function = function
         super().__init__(mobject, **kwargs)
 

manim/animation/transform_matching_parts.py

@@ -20,6 +20,8 @@
 from .transform import FadeTransformPieces, Transform
 
 if TYPE_CHECKING:
+    from typing import Any
+
     from ..scene.scene import Scene
 
 
@@ -74,7 +76,7 @@ def __init__(
         transform_mismatches: bool = False,
         fade_transform_mismatches: bool = False,
         key_map: dict | None = None,
-        **kwargs,
+        **kwargs: Any,
     ):
         if isinstance(mobject, OpenGLVMobject):
             group_type = OpenGLVGroup
@@ -162,11 +164,11 @@ def clean_up_from_scene(self, scene: Scene) -> None:
         scene.add(self.to_add)
 
     @staticmethod
-    def get_mobject_parts(mobject: Mobject):
+    def get_mobject_parts(mobject: Mobject) -> None:
         raise NotImplementedError("To be implemented in subclass.")
 
     @staticmethod
-    def get_mobject_key(mobject: Mobject):
+    def get_mobject_key(mobject: Mobject) -> None:
         raise NotImplementedError("To be implemented in subclass.")
 
 
@@ -206,7 +208,7 @@ def __init__(
         transform_mismatches: bool = False,
         fade_transform_mismatches: bool = False,
         key_map: dict | None = None,
-        **kwargs,
+        **kwargs: Any,
     ):
         super().__init__(
             mobject,
@@ -269,7 +271,7 @@ def __init__(
         transform_mismatches: bool = False,
         fade_transform_mismatches: bool = False,
         key_map: dict | None = None,
-        **kwargs,
+        **kwargs: Any,
     ):
         super().__init__(
             mobject,

manim/camera/camera.py

@@ -274,7 +274,9 @@ def init_background(self):
             )
             self.background[:, :] = background_rgba
 
-    def get_image(self, pixel_array: np.ndarray | list | tuple | None = None):
+    def get_image(
+        self, pixel_array: np.ndarray | list | tuple | None = None
+    ) -> Image.Image:
         """Returns an image from the passed
         pixel array, or from the current frame
         if the passed pixel array is none.

manim/camera/moving_camera.py

@@ -10,6 +10,8 @@
 
 __all__ = ["MovingCamera"]
 
+from typing import TYPE_CHECKING
+
 import numpy as np
 
 from .. import config
@@ -19,6 +21,11 @@
 from ..mobject.mobject import Mobject
 from ..utils.color import WHITE
 
+if TYPE_CHECKING:
+    from typing import Any
+
+    from manim.utils.color import ParsableManimColor
+
 
 class MovingCamera(Camera):
     """
@@ -32,12 +39,12 @@ class MovingCamera(Camera):
 
     def __init__(
         self,
-        frame=None,
-        fixed_dimension=0,  # width
-        default_frame_stroke_color=WHITE,
-        default_frame_stroke_width=0,
-        **kwargs,
-    ):
+        frame: ScreenRectangle | None = None,
+        fixed_dimension: Any = 0,  # width
+        default_frame_stroke_color: ParsableManimColor = WHITE,
+        default_frame_stroke_width: float = 0,
+        **kwargs: Any,
+    ) -> None:
         """
         Frame is a Mobject, (should almost certainly be a rectangle)
         determining which region of space the camera displays

manim/camera/multi_camera.py

@@ -5,7 +5,7 @@
 __all__ = ["MultiCamera"]
 
 
-from manim.mobject.types.image_mobject import ImageMobject
+from manim.mobject.types.image_mobject import ImageMobject, ImageMobjectFromCamera
 
 from ..camera.moving_camera import MovingCamera
 from ..utils.iterables import list_difference_update
@@ -38,7 +38,9 @@ def __init__(
         )
         super().__init__(**kwargs)
 
-    def add_image_mobject_from_camera(self, image_mobject_from_camera: ImageMobject):
+    def add_image_mobject_from_camera(
+        self, image_mobject_from_camera: ImageMobjectFromCamera
+    ):
         """Adds an ImageMobject that's been obtained from the camera
         into the list ``self.image_mobject_from_cameras``
 

manim/camera/three_d_camera.py

@@ -84,7 +84,7 @@ def capture_mobjects(self, mobjects, **kwargs):
         self.reset_rotation_matrix()
         super().capture_mobjects(mobjects, **kwargs)
 
-    def get_value_trackers(self):
+    def get_value_trackers(self) -> list[ValueTracker]:
         """A list of :class:`ValueTrackers <.ValueTracker>` of phi, theta, focal_distance,
         gamma and zoom.
 

manim/gui/gui.py

@@ -11,6 +11,8 @@
 
 
 from .. import __version__, config
+from ..renderer.cairo_renderer import CairoRenderer
+from ..renderer.opengl_renderer import OpenGLRenderer
 from ..utils.module_ops import scene_classes_from_file
 
 __all__ = ["configure_pygui"]
@@ -20,7 +22,9 @@
     window = dpg.generate_uuid()
 
 
-def configure_pygui(renderer, widgets, update=True):
+def configure_pygui(
+    renderer: CairoRenderer | OpenGLRenderer, widgets, update: bool = True
+) -> None:
     if not dearpygui_imported:
         raise RuntimeError("Attempted to use DearPyGUI when it isn't imported.")
     if update:

manim/mobject/geometry/arc.py

@@ -273,22 +273,22 @@ def get_first_handle(self) -> InternalPoint3D:
         # Type inference of extracting an element from a list, is not
         # supported by numpy, see this numpy issue
         # https://github.com/numpy/numpy/issues/16544
-        return self.points[1]
+        return self.points[1]  # type: ignore[no-any-return]
 
     def get_last_handle(self) -> InternalPoint3D:
-        return self.points[-2]
+        return self.points[-2]  # type: ignore[no-any-return]
 
     def get_end(self) -> InternalPoint3D:
         if self.has_tip():
-            return self.tip.get_start()
+            return self.tip.get_start()  # type: ignore[return-value]
         else:
-            return super().get_end()
+            return super().get_end()  # type: ignore[return-value]
 
     def get_start(self) -> InternalPoint3D:
         if self.has_start_tip():
-            return self.start_tip.get_start()
+            return self.start_tip.get_start()  # type: ignore[return-value]
         else:
-            return super().get_start()
+            return super().get_start()  # type: ignore[return-value]
 
     def get_length(self) -> float:
         start, end = self.get_start_and_end()
@@ -322,10 +322,10 @@ def __init__(
         if radius is None:  # apparently None is passed by ArcBetweenPoints
             radius = 1.0
         self.radius = radius
-        self.num_components = num_components
-        self.arc_center = arc_center
-        self.start_angle = start_angle
-        self.angle = angle
+        self.num_components: int = num_components
+        self.arc_center: InternalPoint3D = arc_center
+        self.start_angle: float = start_angle
+        self.angle: float = angle
         self._failed_to_get_center: bool = False
         super().__init__(**kwargs)
 
@@ -484,7 +484,7 @@ def __init__(
         if radius is None:
             center = self.get_arc_center(warning=False)
             if not self._failed_to_get_center:
-                temp_radius: float = np.linalg.norm(np.array(start) - np.array(center))
+                temp_radius: float = np.linalg.norm(np.array(start) - np.array(center))  # type: ignore[assignment]
                 self.radius = temp_radius
             else:
                 self.radius = np.inf
@@ -661,7 +661,7 @@ def construct(self):
             perpendicular_bisector([np.asarray(p1), np.asarray(p2)]),
             perpendicular_bisector([np.asarray(p2), np.asarray(p3)]),
         )
-        radius: float = np.linalg.norm(p1 - center)
+        radius: float = np.linalg.norm(p1 - center)  # type: ignore[assignment]
         return Circle(radius=radius, **kwargs).shift(center)
 
 

manim/mobject/geometry/boolean_ops.py

@@ -5,7 +5,7 @@
 from typing import TYPE_CHECKING
 
 import numpy as np
-from pathops import Path as SkiaPath
+from pathops import Path as SkiaPath  # type: ignore[import-untyped]
 from pathops import PathVerb, difference, intersection, union, xor
 
 from manim import config
@@ -106,7 +106,7 @@ def _convert_vmobject_to_skia_path(self, vmobject: VMobject) -> SkiaPath:
                 for _p0, p1, p2, p3 in quads:
                     path.cubicTo(*p1[:2], *p2[:2], *p3[:2])
 
-                if vmobject.consider_points_equals_2d(subpath[0], subpath[-1]):
+                if vmobject.consider_points_equals_2d(subpath[0], subpath[-1]):  # type: ignore[arg-type]
                     path.close()
 
         return path

manim/mobject/geometry/line.py

@@ -145,9 +145,9 @@ def _pointify(
         if isinstance(mob_or_point, (Mobject, OpenGLMobject)):
             mob = mob_or_point
             if direction is None:
-                return mob.get_center()
+                return mob.get_center()  # type: ignore[return-value]
             else:
-                return mob.get_boundary_point(direction)
+                return mob.get_boundary_point(direction)  # type: ignore[return-value]
         return np.array(mob_or_point)
 
     def set_path_arc(self, new_value: float) -> None:
@@ -156,8 +156,8 @@ def set_path_arc(self, new_value: float) -> None:
 
     def put_start_and_end_on(
         self,
-        start: InternalPoint3D,
-        end: InternalPoint3D,
+        start: InternalPoint3D,  # type: ignore[override]
+        end: InternalPoint3D,  # type: ignore[override]
     ) -> Self:
         """Sets starts and end coordinates of a line.
 
@@ -309,7 +309,7 @@ def get_start(self) -> InternalPoint3D:
             array([-1.,  0.,  0.])
         """
         if len(self.submobjects) > 0:
-            return self.submobjects[0].get_start()
+            return self.submobjects[0].get_start()  # type: ignore[return-value]
         else:
             return super().get_start()
 
@@ -324,7 +324,7 @@ def get_end(self) -> InternalPoint3D:
             array([1., 0., 0.])
         """
         if len(self.submobjects) > 0:
-            return self.submobjects[-1].get_end()
+            return self.submobjects[-1].get_end()  # type: ignore[return-value]
         else:
             return super().get_end()
 

manim/mobject/geometry/polygram.py

@@ -89,7 +89,7 @@ def __init__(
         for vertices in vertex_groups:
             # The inferred type for *vertices is Any, but it should be
             # InternalPoint3D_Array
-            first_vertex, *vertices = vertices
+            first_vertex, *vertices = vertices  # type: ignore[assignment]
             first_vertex = np.array(first_vertex)
 
             self.start_new_path(first_vertex)
@@ -322,7 +322,7 @@ def construct(self):
     """
 
     def __init__(self, *vertices: InternalPoint3D, **kwargs: Any) -> None:
-        super().__init__(vertices, **kwargs)
+        super().__init__(vertices, **kwargs)  # type: ignore[arg-type]
 
 
 class RegularPolygram(Polygram):
@@ -413,7 +413,7 @@ def gen_polygon_vertices(start_angle: float | None) -> tuple[list[Any], float]:
 
             vertex_groups.append(group)
 
-        super().__init__(*vertex_groups, **kwargs)
+        super().__init__(*vertex_groups, **kwargs)  # type: ignore[arg-type]
 
 
 class RegularPolygon(RegularPolygram):

manim/mobject/geometry/tips.py

@@ -152,7 +152,7 @@ def tip_point(self) -> InternalPoint3D:
         # Type inference of extracting an element from a list, is not
         # supported by numpy, see this numpy issue
         # https://github.com/numpy/numpy/issues/16544
-        return self.points[0]
+        return self.points[0]  # type: ignore[no-any-return]
 
     @property
     def vector(self) -> Vector3D:

manim/mobject/matrix.py

@@ -400,7 +400,7 @@ def add_background_to_entries(self):
             mob.add_background_rectangle()
         return self
 
-    def get_mob_matrix(self):
+    def get_mob_matrix(self) -> VGroup:
         """Return the underlying mob matrix mobjects.
 
         Returns
@@ -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