Skip to content

Commit 49d7a52

Browse files
committed
style: black formatting for digamma.py
1 parent 18dcd86 commit 49d7a52

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

  • src/flag_gems/runtime/backend/_kunlunxin/ops

src/flag_gems/runtime/backend/_kunlunxin/ops/digamma.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,9 @@ def _digamma(x):
2020

2121
rinv = 1.0 / r
2222
z = rinv * rinv
23-
poly = z * (1.0 / 12.0 + z * (-1.0 / 120.0 + z * (1.0 / 252.0
24-
+ z * (-1.0 / 240.0))))
23+
poly = z * (
24+
1.0 / 12.0 + z * (-1.0 / 120.0 + z * (1.0 / 252.0 + z * (-1.0 / 240.0)))
25+
)
2526
psi = tl.math.log(r) - 0.5 * rinv - poly - inv_sum
2627

2728
cot = tl.math.cos(PI * x) / tl.math.sin(PI * x)

0 commit comments

Comments
 (0)