code: def ensure_jnp(x):
      """Return a copy of x that is ensured to be jnp.ndarray."""
      return jnp.array(x)
  
imports: from typing import NamedTuple
  from functools import partial
  import logging
  import math
  from typing import Optional
  import numpy
  import jax
  from jax import numpy
  from evojax.algo.base import NEAlgorithm
  from evojax.util import create_logger
label_desc: Use image node_id 1 for calling a global function with example usage: ensure_jnp(x) and returns: jnp