From f2188db4aaa30ba863ed7ea18a1ed8f2c0f1abcb Mon Sep 17 00:00:00 2001 From: Ilia Ilmer Date: Sun, 2 Aug 2026 21:18:34 -0400 Subject: [PATCH] fix --- tinyflow/time_sampler.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tinyflow/time_sampler.py b/tinyflow/time_sampler.py index 07d93fd..9da04e6 100644 --- a/tinyflow/time_sampler.py +++ b/tinyflow/time_sampler.py @@ -42,7 +42,7 @@ def __init__(self, scale: float = 0.5): def sample(self, *shape) -> Tensor: u = Tensor.rand(*shape) - c = (math.pi * u).cos() ** 2 + c = (math.pi / 2 * u).cos() ** 2 return 1 - u - self.s * (c - 1 + u) @@ -52,5 +52,5 @@ def __init__(self): def sample(self, *shape) -> Tensor: u = Tensor.rand(*shape) - t = 1 - 1 / ((math.pi * u).tan() + 1) + t = 1 - 1 / ((math.pi / 2 * u).tan() + 1) return t