Cache is_array_api_object

Author: crusaderky

array_api_compat/common/_helpers.py

@@ -229,10 +229,11 @@ def is_array_api_obj(x: object) -> bool:
     is_dask_array
     is_jax_array
     """
-    if hasattr(x, '__array_namespace__'):
-        return True
+    return hasattr(x, '__array_namespace__') or _is_array_api_cls(type(x))
 
-    cls = type(x)
+
+@cache
+def _is_array_api_cls(cls: type) -> bool:
     return (
         # TODO: drop support for numpy<2 which didn't have __array_namespace__
         _issubclass_fast(cls, "numpy", "ndarray")