Skip to content

Commit c278c32

Browse files
committed
Fix missing docstring.
1 parent 0de7040 commit c278c32

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

src/array_api_extra/_delegation.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,6 @@ def one_hot(
149149
An array having the same shape as `x` except for a new axis at the position
150150
given by `axis` having size `num_classes`.
151151
152-
153152
If ``x < 0`` or ``x >= num_classes``, then the result is undefined, may raise
154153
an exception, or may even cause a bad state. `x` is not checked.
155154

src/array_api_extra/_lib/_funcs.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -388,8 +388,8 @@ def one_hot(
388388
x_size: int,
389389
dtype: DType,
390390
xp: ModuleType,
391-
) -> Array:
392-
"""Helper for _delegation.one_hot."""
391+
) -> Array: # numpydoc ignore=PR01,RT01
392+
"""See docstring in `array_api_extra._delegation.py`."""
393393
out = xp.zeros((x.size, num_classes), dtype=dtype)
394394
x_flattened = xp.reshape(x, (-1,))
395395
if is_numpy_namespace(xp):

0 commit comments

Comments
 (0)