Update documentation and typings for ParametricFunction

manim/mobject/graphing/functions.py

@@ -8,12 +8,14 @@
 from typing import Callable, Iterable, Sequence
 
 import numpy as np
+import numpy.typing as npt
 from isosurfaces import plot_isoline
 
 from manim import config
 from manim.mobject.graphing.scale import LinearBase, _ScaleBase
 from manim.mobject.opengl.opengl_compatibility import ConvertToOpenGL
 from manim.mobject.types.vectorized_mobject import VMobject
+from manim.typing import Point2D, Point3D
 from manim.utils.color import YELLOW
 
 
@@ -23,7 +25,7 @@ class ParametricFunction(VMobject, metaclass=ConvertToOpenGL):
     Parameters
     ----------
     function
-        The function to be plotted in the form of ``(lambda x: x**2)``
+        The function to be plotted in the form of ``(lambda t: (x(t), y(t), z(t)))``
     t_range
         Determines the length that the function spans. By default ``[0, 1]``
     scaling
@@ -97,8 +99,8 @@ def construct(self):
 
     def __init__(
         self,
-        function: Callable[[float, float], float],
-        t_range: Sequence[float] | None = None,
+        function: Callable[[float], Point3D],
+        t_range: Point2D | None = None,
         scaling: _ScaleBase = LinearBase(),
         dt: float = 1e-8,
         discontinuities: Iterable[float] | None = None,