code: def make_ensemble_world_model(
      world_model_network: mbop_networks.WorldModelNetwork,
  ) -> WorldModel:
      """Creates an ensemble world model from its network."""
      return functools.partial(
          ensemble.apply_round_robin, world_model_network.apply
      )
  
file_path: acme/acme/agents/jax/mbop/models.py
imports: import functools
  from typing import Callable, Generic, Optional, Tuple
  from acme import specs
  from acme.agents.jax import actor_core
  from acme.agents.jax.mbop import ensemble
  from acme.agents.jax.mbop import networks
  from acme.jax import networks
  from acme.jax import utils
  import chex
  import jax
label_desc: Use image node_id 2 for calling a global function with example usage: make_ensemble_world_model(world_model_network) and returns: functools