(self)
return {
            "t_prof": self.t_prof,
            "mode": self._mode,
            "env": self._internal_env_wrapper.state_dict(),
            "agent": self._state_dict(),
        }
Override and keep base as one field!
Override and keep base as one field!
Override and keep base as one field!
def state_dict(self):
        """ Override and keep base as one field! """
        return {
            "t_prof": self.t_prof,
            "mode": self._mode,
            "env": self._internal_env_wrapper.state_dict(),
            "agent": self._state_dict(),
        }
def state_dict ( self ) : return { "t_prof" : self . t_prof , "mode" : self . _mode , "env" : self . _internal_env_wrapper . state_dict ( ) , "agent" : self . _state_dict ( ) , }