Skip to content

Commit 23f390e

Browse files
committed
Add a comment about atanh special casing
1 parent 727072f commit 23f390e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

array_api_strict/tests/test_elementwise_functions.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,6 @@ def _array_vals(dtypes):
134134
for func_name, types in elementwise_function_input_types.items():
135135
dtypes = _dtype_categories[types]
136136
func = getattr(_elementwise_functions, func_name)
137-
print(f"{func_name=} {nargs(func)=} {types=} {dtypes=}")
138137

139138
for x in _array_vals(dtypes):
140139
if nargs(func) == 2:
@@ -144,6 +143,8 @@ def _array_vals(dtypes):
144143
assert r.device == x.device
145144

146145
else:
146+
# `atanh` needs a slightly different input value from
147+
# everyone else
147148
if func_name == "atanh":
148149
x -= 0.1
149150
r = func(x)

0 commit comments

Comments
 (0)