def __init__(self, *args, observation_key=None, desired_goal_key=None, **kwargs): HER.__init__( self, observation_key=observation_key, desired_goal_key=desired_goal_key, ) SoftActorCritic.__init__(self, *args, **kwargs) assert isinstance(self.replay_buffer, ObsDictRelabelingBuffer)
def __init__( self, *args, her_kwargs, sac_kwargs, **kwargs ): HER.__init__(self, **her_kwargs) SoftActorCritic.__init__(self, *args, **kwargs, **sac_kwargs) assert isinstance( self.replay_buffer, RelabelingReplayBuffer ) or isinstance( self.replay_buffer, ObsDictRelabelingBuffer )