def save_state_to_checkpoint(self, checkpoint_dir: str, checkpoint_prefix: str):
    self.running_observation_stats.save_state_to_checkpoint(checkpoint_dir, checkpoint_prefix)
def save_state_to_checkpoint(self, checkpoint_dir: str, checkpoint_prefix: str):
    self.running_observation_stats.save_state_to_checkpoint(checkpoint_dir, checkpoint_prefix)