Fix the docstring for kd.index.

odimka · copybara-github · commit dea094962da8 · 2025-01-15T05:39:25.000-08:00
PiperOrigin-RevId: 715761887
Change-Id: Idd2b426334be1ee55bc27d5cca66c6b1e6615cd0
diff --git a/py/koladata/operators/slices.py b/py/koladata/operators/slices.py
@@ -770,7 +770,7 @@ def normalize_dim(x, dim):
     ],
 )
 def index(x, dim=-1):
-  """Returns the indices of the elements computed over the last dim dimensions.
+  """Returns the indices of the elements computed over dimension `dim`.
 
   The resulting slice has the same shape as the input.
 

Original file line number	Diff line number	Diff line change
`@@ -770,7 +770,7 @@ def normalize_dim(x, dim):`
`770`	`770`	`],`
`771`	`771`	`)`
`772`	`772`	`def index(x, dim=-1):`
`773`		`- """Returns the indices of the elements computed over the last dim dimensions.`
	`773`	+ """Returns the indices of the elements computed over dimension `dim`.
`774`	`774`
`775`	`775`	`The resulting slice has the same shape as the input.`
`776`	`776`