cleanup

Genentech · Jun 4, 2024 · 2ca8b82 · 2ca8b82
1 parent b9bf11c
commit 2ca8b82
Showing 1 changed file with 3 additions and 8 deletions.
diff --git 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)