From 2ca8b8220664cf48ae00bb2becda0e275debf2f4 Mon Sep 17 00:00:00 2001 From: Allen Goodman Date: Tue, 4 Jun 2024 17:53:20 -0400 Subject: [PATCH] cleanup --- src/beignet/special/_dawson_integral_f.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/src/beignet/special/_dawson_integral_f.py b/src/beignet/special/_dawson_integral_f.py index cdf8d8d178..07db8de2c6 100644 --- a/src/beignet/special/_dawson_integral_f.py +++ b/src/beignet/special/_dawson_integral_f.py @@ -3,13 +3,8 @@ import torch from torch import Tensor -import beignet.special +from ._error_erfi import error_erfi -def dawson_integral_f(input: Tensor) -> Tensor: - return ( - math.sqrt(torch.pi) - / 2.0 - * torch.exp(-torch.square(input)) - * beignet.special.error_erfi(input) - ) +def dawson_integral_f(z: Tensor) -> Tensor: + return math.sqrt(torch.pi) / 2.0 * torch.exp(-torch.square(z)) * error_erfi(z)