class_docstr: A variable client for updating variables from a remote source.
class_docstr_tok: ["A","variable","client","for","updating","variables","from","a","remote","source","."]
code: def _callback(self, params_list: List[network_types.Params]):
      if self._device and not isinstance(
          self._client, ReferenceVariableSource
      ):
          # Move variables to a proper device.
          self._params = jax.device_put(params_list, self._device)
      else:
          self._params = params_list
  
file_path: acme/acme/jax/variable_utils.py
imports: from concurrent import futures
  import datetime
  import time
  from typing import List, NamedTuple, Optional, Sequence, Union
  from acme import core
  from acme.jax import networks
  import jax
label_desc: Use image node_id 4 for calling the VariableClient obj's underlying member method code with example usage: obj._callback(params_list) without return types