code: def make_networks(
      spec: specs.EnvironmentSpec,
      hidden_layer_sizes: Sequence[int] = (256, 256),
  ) -> PPONetworks:
      if isinstance(spec.actions, specs.DiscreteArray):
          return make_discrete_networks(spec, hidden_layer_sizes)
      else:
          return make_continuous_networks(
              spec,
              policy_layer_sizes=hidden_layer_sizes,
              value_layer_sizes=hidden_layer_sizes,
          )
  
imports: import dataclasses
  from typing import Callable, NamedTuple, Optional, Sequence
  from acme import specs
  from acme.agents.jax import actor_core
  from acme.jax import networks
  from acme.jax import utils
  import haiku
  import jax
  import jax.numpy
  import numpy
  import tensorflow_probability
label_desc: Use image node_id 2 for calling a global function with example usage: make_networks(spec, hidden_layer_sizes) and returns: make_discrete_networks, make_continuous_networks