From 07e62e9f5c19012ccc034946d6e12166929422f1 Mon Sep 17 00:00:00 2001 From: Athan Date: Thu, 12 Dec 2024 02:16:25 -0800 Subject: [PATCH] docs: update description Co-authored-by: ndgrigorian <46709016+ndgrigorian@users.noreply.github.com> --- src/array_api_stubs/_draft/searching_functions.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/array_api_stubs/_draft/searching_functions.py b/src/array_api_stubs/_draft/searching_functions.py index 62afd235a..69c0f195b 100644 --- a/src/array_api_stubs/_draft/searching_functions.py +++ b/src/array_api_stubs/_draft/searching_functions.py @@ -155,7 +155,7 @@ def top_k( mode: Literal["largest", "smallest"] = "largest", ) -> Tuple[array, array]: """ - Returns the ``k`` largest (or smallest) elements of an input array ``x`` along a specified dimension. + Returns the values and indices of the ``k`` largest (or smallest) elements of an input array ``x`` along a specified dimension. Parameters ----------