buffer error

ml-agents/mlagents/trainers/buffer.py

             if key not in self.keys():
                 return False
             if (length is not None) and (length != len(self[key])):
+                print(length, key, len(self[key]))
                 return False
             length = len(self[key])
         return True

ml-agents/mlagents/trainers/optimizer/torch_optimizer.py

         if done:
             for k in next_value_estimates:
                 if not self.reward_signals[k].ignore_done:
-                    next_value_estimates[k] = np.array([0])
+                    next_value_estimates[k][-1] = 0.0
 
         return (
             value_estimates,