code: def _get_model1(
      variable_source: core.VariableSource,
  ) -> types.ModelToSnapshot:
      return types.ModelToSnapshot(
          model=_model1,
          params=variable_source.get_variables(["params_model1"])[0],
          dummy_kwargs={
              "x": jnp.ones([3, 1], dtype=jnp.float32),
          },
      )
  
imports: import os
  import time
  from typing import Any, Sequence
  from acme import core
  from acme.jax import snapshotter
  from acme.jax import types
  from acme.testing import test_utils
  import jax.numpy
  from absl.testing import absltest
label_desc: Use image node_id 4 for calling a global function with example usage: _get_model1(variable_source) and returns: types