def METHOD_NAME(self, actor_critic, policy_version, device: torch.device):
    self.actor_critic = actor_critic
    self.policy_versions[self.policy_id] = policy_version
    self.device = device
    log.debug("Initialized policy %d weights for model version %d", self.policy_id, policy_version)