f METHOD_NAME(
  self,
  random_key: networks_lib.PRNGKey,
  policy: acting.ActorCore,
  environment_spec: specs.EnvironmentSpec,
  variable_source: Optional[core.VariableSource] = None,
-> core.Actor: