Review edits

Author: NeilGirdhar

src/array_api_extra/_lib/_funcs.py

@@ -386,7 +386,7 @@ def one_hot(
     xp: ModuleType,
 ) -> Array:  # numpydoc ignore=PR01,RT01
     """See docstring in `array_api_extra._delegation.py`."""
-    x_size = x.size
+    x_size = _compat.size(x)
     if x_size is None:  # pragma: no cover
         # This cannot be tested because there is no way to create an array with abstract
         # size today.  However, it is blocked for the sake of type-checking and