[RLlib] - Fix TensorType (#55694)

Author: simonsays1980

rllib/utils/typing.py

@@ -39,8 +39,8 @@
         jnp = jax.numpy
 
 # Represents a generic tensor type.
-# This could be an np.ndarray, tf.Tensor, or a torch.Tensor.
-TensorType = Union[np.array, "jnp.ndarray", "tf.Tensor", "torch.Tensor"]
+# This could be an np.ndarray, jnp.ndarray, tf.Tensor, or a torch.Tensor.
+TensorType = Union[np.ndarray, "jnp.ndarray", "tf.Tensor", "torch.Tensor"]
 
 # Either a plain tensor, or a dict or tuple of tensors (or StructTensors).
 TensorStructType = Union[TensorType, dict, tuple]