def __init__(self, actor_model, critic_model):
    self.actor_model = actor_model
    self.critic_model = critic_model
def __init__(self, actor_model, critic_model):
    self.actor_model = actor_model
    self.critic_model = critic_model