Method append

src/pymgrid/microgrid/utils/step.py:13–31

(self, module_name, obs, reward, done, info)

        self._info = dict(absorbed_energy=[], provided_energy=[])

    def append(self, module_name, obs, reward, done, info):
        try:
            self._obs[module_name].append(obs)
        except KeyError:
            self._obs[module_name] = [obs]
        self._reward += reward
        if done:
            self._done = True

        try:
            self._info[module_name].append(info)
        except KeyError:
            self._info[module_name] = [info]

        for key, value in info.items():
            try:
                self._info[key].append(value)
            except KeyError:
                pass

    def balance(self, shape_reward=False):
        provided_energy = np.sum(self._info['provided_energy'])