Function new_explore_step

tests/manager/synchronizer_test.py:52–63

(self: Explorer)

def explorer_monkey_patch(explore_step_time_list: List[int]):
    async def new_explore_step(self: Explorer):
        if self.explore_step_num >= len(explore_step_time_list):
            await self.finish_current_steps()
            await self.save_checkpoint()
            await self.synchronizer.set_explorer_status.remote(
                RunningStatus.STOPPED,
                old_status=RunningStatus.RUNNING,
            )
            await self.shutdown()
            return False
        self.explore_step_num += 1
        return True

    async def new_finish_explore_step(self: Explorer, step: int) -> None:
        metric = {"rollout/model_version": self.model_version}