memory.py

def update_memory(agent):
    trace = {
        "Ξ": round(agent.Xi["Ξ"], 4),
        "Ψ": round(agent.Psi["Ψ"], 4)
    }
    agent.Omega.append(trace)