code: def make_network(action_spec: specs.DiscreteArray) -> snt.Module:
      return snt.Sequential(
          [
              lambda x: tf.image.convert_image_dtype(x, tf.float32),
              networks.DQNAtariNetwork(action_spec.num_values),
          ]
      )
  
file_path: acme/examples/offline/run_bcq.py
imports: from absl import app
  from absl import flags
  import acme
  from acme import specs
  from acme.agents.tf import actors
  from acme.agents.tf import bcq
  from acme.tf import networks
  from acme.tf import utils
  from acme.utils import counting
  from acme.utils import loggers
  import sonnet
  import tensorflow
  import trfl
  from deepmind_research.rl_unplugged import atari
label_desc: Use image node_id 1 for calling a global function with example usage: make_network(action_spec) and returns: snt