Label NxGraph as a TypeAlias

ManimCommunity · Feb 5, 2024 · 4ce3fe0 · 4ce3fe0
1 parent f32985d
commit 4ce3fe0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/manim/mobject/graph.py b/manim/mobject/graph.py
@@ -10,6 +10,7 @@
 import itertools as it
 from copy import copy
 from typing import Any, Hashable, Iterable, Literal, Protocol, Union, cast
+from typing_extensions import TypeAlias
 
 import networkx as nx
 import numpy as np
@@ -26,7 +27,7 @@
 from manim.typing import Point3D
 from manim.utils.color import BLACK
 
-NxGraph = Union[nx.classes.graph.Graph, nx.classes.digraph.DiGraph]
+NxGraph: TypeAlias = Union[nx.classes.graph.Graph, nx.classes.digraph.DiGraph]
 
 
 class LayoutFunction(Protocol):