code: def make_networks(
      spec: specs.EnvironmentSpec, **kwargs
  ) -> CQLNetworks:
      sac_networks = sac.make_networks(spec, **kwargs)
      return CQLNetworks(
          policy_network=sac_networks.policy_network,
          critic_network=sac_networks.q_network,
          log_prob=sac_networks.log_prob,
          sample=sac_networks.sample,
          sample_eval=sac_networks.sample_eval,
          environment_specs=spec,
      )
  
file_path: acme/acme/agents/jax/cql/networks.py
imports: import dataclasses
  from typing import Optional, Tuple
  from acme import specs
  from acme.agents.jax import sac
  from acme.jax import networks
  import jax
  import jax.numpy
label_desc: Use image node_id 2 for calling a global function with example usage: make_networks(spec) and returns: CQLNetworks