cleanup

Genentech · Jun 4, 2024 · f1df9d4 · f1df9d4
1 parent 2ca8b82
commit f1df9d4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/beignet/special/_dawson_integral_f.py b/src/beignet/special/_dawson_integral_f.py
@@ -6,5 +6,5 @@
 from ._error_erfi import error_erfi
 
 
-def dawson_integral_f(z: Tensor) -> Tensor:
-    return math.sqrt(torch.pi) / 2.0 * torch.exp(-torch.square(z)) * error_erfi(z)
+def dawson_integral_f(input: Tensor) -> Tensor:
+    return math.sqrt(torch.pi) / 2.0 * torch.exp(-(input**2)) * error_erfi(input)