fix curiosity-driven notebook heat plot