Remove unnecessary error control

Author: crusaderky

array_api_compat/common/_helpers.py

@@ -909,8 +909,7 @@ def _common(
         self,
         at_op: str,
         y=_undef,
-        copy: bool | None = True, 
-        mode: str = "promise_in_bounds", 
+        copy: bool | None = True,
         **kwargs,
     ):
         """Validate kwargs and perform common prepocessing.
@@ -931,19 +930,14 @@ def _common(
                 "    at(x)[idx].set(value)\n"
                 "(same for all other methods)."
             )
-        if mode != "promise_in_bounds" and not is_jax_array(self.x):
-            xp = array_namespace(self.x)
-            raise NotImplementedError(
-                f"mode='{mode!r}' is not supported for backend {xp.__name__}"
-            )
 
         copy = _parse_copy_param(self.x, copy)
 
         if copy and is_jax_array(self.x):
             # Use JAX's at[]
             at_ = self.x.at[self.idx]
             args = (y, ) if y is not _undef else ()
-            return getattr(at_, at_op)(*args, mode=mode, **kwargs), None
+            return getattr(at_, at_op)(*args, **kwargs), None
 
         # Emulate at[] behaviour for non-JAX arrays
         x = self.x.copy() if copy else self.x