code: def _fetch_devicearray(x):
      if isinstance(x, jax.Array):
          return np.asarray(x)
      return x
  
file_path: acme/acme/agents/jax/mpo/utils.py
imports: from typing import Callable
  from acme import types
  from acme.adders import reverb
  from acme.agents.jax.mpo import types
  import distrax
  import jax
  import jax.numpy
  import numpy
  import tensorflow_probability.substrates.jax
label_desc: Use image node_id 1 for calling a global function with example usage: _fetch_devicearray(x) and returns: x, np