f METHOD_NAME(
  self,
  random_key: networks_lib.PRNGKey,
  networks: mbop_networks.MBOPNetworks,
  dataset: Iterator[types.Transition],
  logger_fn: loggers.LoggerFactory,
  environment_spec: specs.EnvironmentSpec,
  counter: Optional[counting.Counter] = None,
-> core.Learner: