Imported from https://github.com/ShibiHe/Model-Free-Episodic-Control