code: def make_networks(
      spec: specs.EnvironmentSpec,
      hidden_layer_sizes: Tuple[int, ...] = (64, 64),
  ) -> MBOPNetworks:
      """Creates networks used by the agent."""
      world_model_network = make_world_model_network(
          spec, hidden_layer_sizes=hidden_layer_sizes
      )
      policy_prior_network = make_policy_prior_network(
          spec, hidden_layer_sizes=hidden_layer_sizes
      )
      n_step_return_network = make_n_step_return_network(
          spec, hidden_layer_sizes=hidden_layer_sizes
      )
  
      return MBOPNetworks(
          world_model_network=world_model_network,
          policy_prior_network=policy_prior_network,
          n_step_return_network=n_step_return_network,
      )
  
file_path: acme/acme/agents/jax/mbop/networks.py
imports: import dataclasses
  from typing import Any, Tuple
  from acme import specs
  from acme.jax import networks
  from acme.jax import utils
  import haiku
  import jax.numpy
  import numpy
label_desc: Use image node_id 5 for calling a global function with example usage: make_networks(spec, hidden_layer_sizes) and returns: MBOPNetworks