f METHOD_NAME(
  self,
  random_key: networks_lib.PRNGKey,
  networks: networks_lib.FeedForwardNetwork,
  dataset: Iterator[reverb.ReplaySample],
  logger_fn: loggers.LoggerFactory,
  environment_spec: specs.EnvironmentSpec,
  replay_client: Optional[reverb.Client] = None,
  counter: Optional[counting.Counter] = None,
-> core.Learner: