9128a4b20336fb8c07fc4e0e0985fe299d1a5f12,tensorforce/core/memories/replay_memory.py,ReplayMemory,get_batch,#ReplayMemory#Any#,71

Before Change

            batch["states"][state] = self.states[state].take(indices)
        for action in self.actions:
            batch["actions"][action] = self.actions[action].take(indices)
        batch["rewards"] = self.rewards.take(indices)
        batch["terminals"] = self.rewards.take(indices)
        for n in range(len(self.internals)):
            batch["internals"].append(self.internals.take(indices))

After Change

            actions={name: action.take(indices) for name, action in self.actions.items()},
            rewards=self.rewards.take(indices),
            terminals=self.terminals.take(indices),
            internals=[internal.take(indices, axis=0) for internal in self.internals]
        )

Instances