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