def coordinator(self, counter: counting.Counter, max_learner_steps: int):
return lp_utils.StepsLimiter(counter,
max_steps=max_learner_steps,
steps_key='learner_steps')
def coordinator(self, counter, max_actor_steps):
if self._builder._config.env_name.startswith('offline_ant'): # pytype: disable=attribute-error, pylint: disable=protected-access
steps_key = 'learner_steps'
else:
steps_key = 'actor_steps'
return lp_utils.StepsLimiter(counter, max_actor_steps, steps_key=steps_key)
def coordinator(self, counter: counting.Counter, max_actor_steps: int):
return lp_utils.StepsLimiter(counter, max_actor_steps)