import numpy as np class buffer(object): def __init__(self): self.states = [] self.actions = [] self.rewards = [] print("√√√√√Buffer Initialized Success√√√√√") def clearBuffer(self): self.states = [] self.actions = [] self.rewards = [] def getStates(self): return np.asarray(self.states) def getActions(self): return np.asarray(self.actions) def getRewards(self): return np.asarray(self.rewards) def saveState(self,state): self.states.append(state) def saveAction(self,action): self.actions.append(action) def saveReward(self,reward): self.rewards.append(reward) def saveBuffers(self,state,action,reward): self.states.append(state) self.actions.append(action) self.rewards.append(reward)