Replace uses of jnp.array in types with jnp.ndarray.

`jnp.array` is a function, not a type:
https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.array.html
so it never makes sense to use `jnp.array` in a type annotation. Presumably the intent was to write `jnp.ndarray` aka `jax.Array`.

PiperOrigin-RevId: 559395698

scenic/projects/baselines/detr/detr_base_model.py

@@ -160,9 +160,9 @@ def compute_cost_matrix(self, predictions: ArrayDict,
     """
     raise NotImplementedError('Subclasses must implement compute_cost_matrix.')
 
-  def matcher(self,
-              cost: jnp.ndarray,
-              n_cols: Optional[jnp.array] = None) -> jnp.ndarray:
+  def matcher(
+      self, cost: jnp.ndarray, n_cols: Optional[jnp.ndarray] = None
+  ) -> jnp.ndarray:
     """Implements a matching function.
 
     Matching function matches output detections against ground truth detections