This is the code for First return then explore, the new Go-explore paper. Code for the original paper can be found in this repository under the tag "v1.0" or the release "Go-Explore v1".
The code for Go-Explore with a deterministic exploration phase followed by a robustification phase is located in the robustified subdirectory. The code for Go-Explore with a policy-based exploration phase is located in the policy_based subdirectory. Installation instructions for each variant of Go-Explore can be found in their respective directories.