bug fix. Make_batch method now returns proper value for s_prime

seungeunrho · Jun 24, 2019 · cabbd84 · cabbd84
1 parent 46f9b32
commit cabbd84
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/actor_critic.py b/actor_critic.py
@@ -45,7  45,7 @@ def make_batch(self):
             done_lst.append([done_mask])
 
         s_batch, a_batch, r_batch, s_prime_batch, done_batch = torch.tensor(s_lst, dtype=torch.float), torch.tensor(a_lst), \
-                                                               torch.tensor(r_lst, dtype=torch.float), torch.tensor(s_prime, dtype=torch.float), \
                                                                torch.tensor(r_lst, dtype=torch.float), torch.tensor(s_prime_lst, dtype=torch.float), \
                                                                torch.tensor(done_lst, dtype=torch.float)
         self.data = []
         return s_batch, a_batch, r_batch, s_prime_batch, done_batch