Added type annotations to mobject/svg/brace.py

manim/mobject/svg/brace.py

@@ -5,10 +5,11 @@
 __all__ = ["Brace", "BraceLabel", "ArcBrace", "BraceText", "BraceBetweenPoints"]
 
 from collections.abc import Sequence
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
 
 import numpy as np
 import svgelements as se
+from typing_extensions import Self
 
 from manim._config import config
 from manim.mobject.geometry.arc import Arc
@@ -17,6 +18,7 @@
 from manim.mobject.opengl.opengl_compatibility import ConvertToOpenGL
 from manim.mobject.text.tex_mobject import MathTex, Tex
 
+from ...animation.animation import Animation
 from ...animation.composition import AnimationGroup
 from ...animation.fading import FadeIn
 from ...animation.growing import GrowFromCenter
@@ -26,7 +28,7 @@
 from ..svg.svg_mobject import VMobjectFromSVGPath
 
 if TYPE_CHECKING:
-    from manim.typing import Point3DLike, Vector3D
+    from manim.typing import Point3D, Point3DLike, Vector3D
     from manim.utils.color.core import ParsableManimColor
 
 __all__ = ["Brace", "BraceBetweenPoints", "BraceLabel", "ArcBrace"]
@@ -70,14 +72,14 @@ def construct(self):
     def __init__(
         self,
         mobject: Mobject,
-        direction: Vector3D | None = DOWN,
+        direction: Vector3D = DOWN,
         buff: float = 0.2,
         sharpness: float = 2,
         stroke_width: float = 0,
         fill_opacity: float = 1.0,
         background_stroke_width: float = 0,
         background_stroke_color: ParsableManimColor = BLACK,
-        **kwargs,
+        **kwargs: Any,
     ):
         path_string_template = (
             "m0.01216 0c-0.01152 0-0.01216 6.103e-4 -0.01216 0.01311v0.007762c0.06776 "
@@ -130,7 +132,7 @@ def __init__(
         for mob in mobject, self:
             mob.rotate(angle, about_point=ORIGIN)
 
-    def put_at_tip(self, mob: Mobject, use_next_to: bool = True, **kwargs):
+    def put_at_tip(self, mob: Mobject, use_next_to: bool = True, **kwargs: Any) -> Self:
         """Puts the given mobject at the brace tip.
 
         Parameters
@@ -153,7 +155,7 @@ def put_at_tip(self, mob: Mobject, use_next_to: bool = True, **kwargs):
             mob.shift(self.get_direction() * shift_distance)
         return self
 
-    def get_text(self, *text, **kwargs):
+    def get_text(self, *text: str, **kwargs: Any) -> Tex:
         """Places the text at the brace tip.
 
         Parameters
@@ -172,7 +174,7 @@ def get_text(self, *text, **kwargs):
         self.put_at_tip(text_mob, **kwargs)
         return text_mob
 
-    def get_tex(self, *tex, **kwargs):
+    def get_tex(self, *tex: str, **kwargs: Any) -> MathTex:
         """Places the tex at the brace tip.
 
         Parameters
@@ -191,15 +193,15 @@ def get_tex(self, *tex, **kwargs):
         self.put_at_tip(tex_mob, **kwargs)
         return tex_mob
 
-    def get_tip(self):
+    def get_tip(self) -> Point3D:
         """Returns the point at the brace tip."""
         # Returns the position of the seventh point in the path, which is the tip.
         if config["renderer"] == "opengl":
             return self.points[34]
 
         return self.points[28]  # = 7*4
 
-    def get_direction(self):
+    def get_direction(self) -> Vector3D:
         """Returns the direction from the center to the brace tip."""
         vect = self.get_tip() - self.get_center()
         return vect / np.linalg.norm(vect)
@@ -238,7 +240,7 @@ def __init__(
         font_size: float = DEFAULT_FONT_SIZE,
         buff: float = 0.2,
         brace_config: dict | None = None,
-        **kwargs,
+        **kwargs: Any,
     ):
         self.label_constructor = label_constructor
         super().__init__(**kwargs)
@@ -249,37 +251,51 @@ def __init__(
         self.brace = Brace(obj, brace_direction, buff, **brace_config)
 
         if isinstance(text, (tuple, list)):
-            self.label = self.label_constructor(*text, font_size=font_size, **kwargs)
+            self.label: VMobject = self.label_constructor(
+                *text, font_size=font_size, **kwargs
+            )
         else:
             self.label = self.label_constructor(str(text), font_size=font_size)
 
         self.brace.put_at_tip(self.label)
         self.add(self.brace, self.label)
 
-    def creation_anim(self, label_anim=FadeIn, brace_anim=GrowFromCenter):
+    def creation_anim(
+        self,
+        label_anim: type[Animation] = FadeIn,
+        brace_anim: type[Animation] = GrowFromCenter,
+    ) -> AnimationGroup:
         return AnimationGroup(brace_anim(self.brace), label_anim(self.label))
 
-    def shift_brace(self, obj, **kwargs):
+    # TODO: This method is only called from the method change_brace_label, which is
+    # not called from any location in the current codebase.
+    # TODO: The Mobject could be more specific.
+    def shift_brace(self, obj: Mobject, **kwargs: Any) -> Self:
         if isinstance(obj, list):
             obj = self.get_group_class()(*obj)
         self.brace = Brace(obj, self.brace_direction, **kwargs)
         self.brace.put_at_tip(self.label)
         return self
 
-    def change_label(self, *text, **kwargs):
+    # TODO: This method is only called from the method change_brace_label, which is
+    # not called from any location in the current codebase.
+    def change_label(self, *text: str, **kwargs: Any) -> Self:
         self.label = self.label_constructor(*text, **kwargs)
 
         self.brace.put_at_tip(self.label)
         return self
 
-    def change_brace_label(self, obj, *text, **kwargs):
+    # TODO: This method is not called from anywhere in the codebase.
+    def change_brace_label(self, obj: Mobject, *text: str, **kwargs: Any) -> Self:
         self.shift_brace(obj)
         self.change_label(*text, **kwargs)
         return self
 
 
 class BraceText(BraceLabel):
-    def __init__(self, obj, text, label_constructor=Tex, **kwargs):
+    def __init__(
+        self, obj: Mobject, text: str, label_constructor: type[Tex] = Tex, **kwargs: Any
+    ):
         super().__init__(obj, text, label_constructor=label_constructor, **kwargs)
 
 
@@ -320,7 +336,7 @@ def __init__(
         point_1: Point3DLike | None,
         point_2: Point3DLike | None,
         direction: Vector3D | None = ORIGIN,
-        **kwargs,
+        **kwargs: Any,
     ):
         if all(direction == ORIGIN):
             line_vector = np.array(point_2) - np.array(point_1)
@@ -387,7 +403,7 @@ def __init__(
         self,
         arc: Arc | None = None,
         direction: Sequence[float] = RIGHT,
-        **kwargs,
+        **kwargs: Any,
     ):
         if arc is None:
             arc = Arc(start_angle=-1, angle=2, radius=1)

manim/mobject/text/tex_mobject.py

@@ -29,6 +29,7 @@
 from collections.abc import Iterable
 from functools import reduce
 from textwrap import dedent
+from typing import Any
 
 from manim import config, logger
 from manim.constants import *
@@ -447,7 +448,11 @@ class Tex(MathTex):
     """
 
     def __init__(
-        self, *tex_strings, arg_separator="", tex_environment="center", **kwargs
+        self,
+        *tex_strings: str,
+        arg_separator: str = "",
+        tex_environment: str = "center",
+        **kwargs: Any,
     ):
         super().__init__(
             *tex_strings,

mypy.ini

@@ -153,9 +153,6 @@ ignore_errors = True
 [mypy-manim.mobject.opengl.opengl_vectorized_mobject]
 ignore_errors = True
 
-[mypy-manim.mobject.svg.brace]
-ignore_errors = True
-
 [mypy-manim.mobject.svg.svg_mobject]
 ignore_errors = True