Merge branch 'dev' of github.com:MushroomRL/mushroom-rl into dev

MushroomRL · Jan 22, 2024 · 861451c · 861451c
2 parents ba0ee99 + 0c68434
commit 861451c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mushroom_rl/algorithms/actor_critic/deep_actor_critic/ppo_bptt.py b/mushroom_rl/algorithms/actor_critic/deep_actor_critic/ppo_bptt.py
@@ -126,8 +126,8 @@ def _transform_to_sequences(self, states_old, states, policy_states, actions, ne
                 # apply padding
                 length_seq = len(states_seq)
                 padded_states_old = torch.concatenate([states_old_seq,
-                                                       torch.zeros((self._truncation_length - states_seq.shape[0],
-                                                                    states_seq.shape[1]))])
+                                                       torch.zeros((self._truncation_length - states_old_seq.shape[0],
+                                                                    states_old_seq.shape[1]))])
                 padded_states = torch.concatenate([states_seq,
                                                    torch.zeros((self._truncation_length - states_seq.shape[0],
                                                                 states_seq.shape[1]))])