cap

Author: NeilGirdhar

src/array_api_extra/_lib/_funcs.py

@@ -392,7 +392,7 @@ def one_hot(
         # specification.
         msg = "x must have a concrete size."
         raise TypeError(msg)
-    # TODO: Benchmark whether this is faster on the Numpy backend:
+    # TODO: Benchmark whether this is faster on the NumPy backend:
     # x_flattened = xp.reshape(x, (-1,))
     # out = xp.zeros((x.size, num_classes), dtype=dtype, device=_compat.device(x))
     # out = at(out)[xp.arange(x_size), x_flattened].set(1)