code: def make_ensemble(
      base_network: networks.FeedForwardNetwork,
      ensemble_apply: Callable[..., Any],
      num_networks: int,
  ) -> networks.FeedForwardNetwork:
      return networks.FeedForwardNetwork(
          init=functools.partial(
              ensemble_init, base_network.init, num_networks
          ),
          apply=functools.partial(ensemble_apply, base_network.apply),
      )
  
file_path: acme/acme/agents/jax/mbop/ensemble.py
imports: import functools
  from typing import Any, Callable
  from acme.jax import networks
  import jax
  import jax.numpy
label_desc: Use image node_id 7 for calling a global function with example usage: make_ensemble(base_network, ensemble_apply, num_networks) and returns: networks