Define ManimColor type in constants

manimlib/animation/indication.py

@@ -34,13 +34,9 @@ from manimlib.utils.rate_functions import wiggle
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from colour import Color
+    from manimlib.constants import ManimColor
-    from typing import Union
-
     from manimlib.mobject.mobject import Mobject
 
-    ManimColor = Union[str, Color]
-
 
 class FocusOn(Transform):
     CONFIG = {

manimlib/animation/transform.py

@@ -19,14 +19,10 @@ from manimlib.utils.rate_functions import squish_rate_func
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from colour import Color
+    from typing import Callable
-    from typing import Callable, Union
-
     import numpy.typing as npt
-
     from manimlib.scene.scene import Scene
-
+    from manimlib.constants import ManimColor
-    ManimColor = Union[str, Color]
 
 
 class Transform(Animation):

manimlib/constants.py

@@ -84,6 +84,13 @@ CTRL_MODIFIER = 2
 COMMAND_MODIFIER = 64
 
 # Colors
+from typing import TYPE_CHECKING
+if TYPE_CHECKING:
+    from typing import Union
+    from colour import Color
+    ManimColor = Union[str, Color, None]
+
+
 BLUE_E = "#1C758A"
 BLUE_D = "#29ABCA"
 BLUE_C = "#58C4DD"

manimlib/mobject/coordinate_systems.py

@@ -33,13 +33,11 @@ from manimlib.utils.space_ops import normalize
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from colour import Color
+    from typing import Callable, Iterable, Sequence, Type, TypeVar
-    from typing import Callable, Iterable, Sequence, Type, TypeVar, Union
-
     from manimlib.mobject.mobject import Mobject
+    from manimlib.constants import ManimColor
 
     T = TypeVar("T", bound=Mobject)
-    ManimColor = Union[str, Color]
 
 
 EPSILON = 1e-8

manimlib/mobject/geometry.py

@@ -30,10 +30,8 @@ from manimlib.utils.space_ops import rotation_matrix_transpose
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from colour import Color
+    from typing import Iterable
-    from typing import Iterable, Union
+    from manimlib.constants import ManimColor
-
-    ManimColor = Union[str, Color]
 
 
 DEFAULT_DOT_RADIUS = 0.08

manimlib/mobject/matrix.py

@@ -18,14 +18,9 @@ from manimlib.mobject.types.vectorized_mobject import VMobject
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from colour import Color
-    from typing import Union
-
     import numpy.typing as npt
-
     from manimlib.mobject.mobject import Mobject
-
+    from manimlib.constants import ManimColor
-    ManimColor = Union[str, Color]
 
 
 VECTOR_LABEL_SCALE_FACTOR = 0.8

manimlib/mobject/mobject.py

@@ -47,7 +47,6 @@ from manimlib.utils.space_ops import rotation_matrix_transpose
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from colour import Color
     from typing import Callable, Iterable, Sequence, Union
 
     import numpy.typing as npt
@@ -55,7 +54,7 @@ if TYPE_CHECKING:
     TimeBasedUpdater = Callable[["Mobject", float], None]
     NonTimeUpdater = Callable[["Mobject"], None]
     Updater = Union[TimeBasedUpdater, NonTimeUpdater]
-    ManimColor = Union[str, Color]
+    from manimlib.constants import ManimColor
 
 
 class Mobject(object):
@@ -333,7 +332,7 @@ class Mobject(object):
             parent.assemble_family()
         return self
 
-    def get_family(self, recurse: bool = True):
+    def get_family(self, recurse: bool = True) -> list[Mobject]:
         if recurse:
             return self.family
         else:

manimlib/mobject/probability.py

@@ -18,10 +18,8 @@ from manimlib.utils.iterables import listify
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from colour import Color
+    from typing import Iterable
-    from typing import Iterable, Union
+    from manimlib.constants import ManimColor
-
-    ManimColor = Union[str, Color]
 
 
 EPSILON = 0.0001

manimlib/mobject/shape_matchers.py

@@ -15,11 +15,8 @@ from manimlib.utils.customization import get_customization
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from typing import Union
-
     from manimlib.mobject.mobject import Mobject
-
+    from manimlib.constants import ManimColor
-    ManimColor = Union[str, Color]
 
 
 class SurroundingRectangle(Rectangle):

manimlib/mobject/svg/mtex_mobject.py

@@ -9,13 +9,12 @@ from manimlib.utils.tex_file_writing import tex_content_to_svg_file
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from colour import Color
     import re
     from typing import Iterable, Union
 
     from manimlib.mobject.types.vectorized_mobject import VGroup
+    from manimlib.constants import ManimColor
 
-    ManimColor = Union[str, Color]
     Span = tuple[int, int]
     Selector = Union[
         str,

manimlib/mobject/svg/string_mobject.py

@@ -17,10 +17,9 @@ from manimlib.utils.config_ops import digest_config
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from colour import Color
     from typing import Callable, Iterable, Union
+    from manimlib.constants import ManimColor
 
-    ManimColor = Union[str, Color]
     Span = tuple[int, int]
     Selector = Union[
         str,

manimlib/mobject/svg/tex_mobject.py

@@ -18,10 +18,8 @@ from manimlib.utils.tex_file_writing import tex_content_to_svg_file
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from colour import Color
+    from typing import Iterable
-    from typing import Iterable, Union
+    from manimlib.constants import ManimColor
-
-    ManimColor = Union[str, Color]
 
 
 SCALE_FACTOR_PER_FONT_POINT = 0.001

manimlib/mobject/svg/text_mobject.py

@@ -23,12 +23,10 @@ from manimlib.utils.simple_functions import hash_string
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from colour import Color
     from typing import Iterable, Union
 
     from manimlib.mobject.types.vectorized_mobject import VGroup
-
+    from manimlib.constants import ManimColor
-    ManimColor = Union[str, Color]
     Span = tuple[int, int]
     Selector = Union[
         str,

manimlib/mobject/types/point_cloud_mobject.py

@@ -13,12 +13,9 @@ from manimlib.utils.iterables import resize_with_interpolation
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from colour import Color
+    from typing import Callable
-    from typing import Callable, Union
-
     import numpy.typing as npt
-
+    from manimlib.constants import ManimColor
-    ManimColor = Union[str, Color]
 
 
 class PMobject(Mobject):

manimlib/mobject/types/vectorized_mobject.py

@@ -42,12 +42,10 @@ from manimlib.shader_wrapper import ShaderWrapper
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from colour import Color
+    from typing import Callable, Iterable, Sequence
-    from typing import Callable, Iterable, Sequence, Union
 
     import numpy.typing as npt
-
+    from manimlib.constants import ManimColor
-    ManimColor = Union[str, Color]
 
 
 class VMobject(Mobject):

manimlib/utils/color.py

@@ -13,9 +13,8 @@ from manimlib.utils.iterables import resize_with_interpolation
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from typing import Iterable, Union
+    from typing import Iterable
-
+    from manimlib.constants import ManimColor
-    ManimColor = Union[str, Color]
 
 
 def color_to_rgb(color: ManimColor) -> np.ndarray: