public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-21 05:44:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Tweak to type hints

Browse source
This commit is contained in:
Grant Sanderson 2023-01-18 16:25:32 -08:00
parent 8e2cf04b71
commit 3820e098c0
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
manimlib/utils/bezier.py
View file

@ -22,14 +22,14 @@ CLOSED_THRESHOLD = 0.001
def bezier( def bezier(
points: Sequence[Scalable] | VectNArray points: Sequence[float | FloatArray] | VectNArray
) -> Callable[[float], Scalable]: ) -> Callable[[float], float | FloatArray]:
if len(points) == 0: if len(points) == 0:
raise Exception("bezier cannot be calld on an empty list") raise Exception("bezier cannot be calld on an empty list")
n = len(points) - 1 n = len(points) - 1
def result(t: float) -> Scalable: def result(t: float) -> float | FloatArray:
return sum( return sum(
((1 - t)**(n - k)) * (t**k) * choose(n, k) * point ((1 - t)**(n - k)) * (t**k) * choose(n, k) * point
for k, point in enumerate(points) for k, point in enumerate(points)