memory.py

import numpy as np
from collections import deque

import config

class Memory:
    def __init__(self, MEMORY_SIZE):
        self.MEMORY_SIZE = config.MEMORY_SIZE
        self.ltmemory = deque(maxlen=config.MEMORY_SIZE)
        self.stmemory = deque(maxlen=config.MEMORY_SIZE)

    def commit_stmemory(self, identities, state, actionValues):
        for r in identities(state, actionValues):
            self.stmemory.append({'board': r[0].history,
                                  'state': r[0],
                                  'id': r[0].id,
                                  'AV': r[1],
                                  'playerTurn': r[0].playerTurn})

    def commit_ltmemory(self):
        for i in self.stmemory:
            self.ltmemory.append(i)
        self.clear_stmemory()

    def clear_stmemory(self):
        self.stmemory = deque(maxlen=config.MEMORY_SIZE)