__init__() missing 2 required positional arguments: 'env' and 'cell_factory'

[robjlyons]

Hi, I have run this in google colab, just running through your ReadMe and got this error.

from go_explore import GoExplore

go_explore = GoExplore("MontezumaRevenge-v0")
go_explore.explore(total_timesteps=100000)
go_explore.robustify(total_timesteps=1000)

TypeError: __init__() missing 2 required positional arguments: 'env' and 'cell_factory'

[qgallouedec]

Thank you for reporting.
I am currently working on this repository. It is not fully functional yet. Contributions are welcome.
For the moments, the exploration seems to work:

from go_explore import GoExplore
from go_explore.cells import ImageGrayscaleDownscale
from stable_baselines3 import DQN

cell_factory = ImageGrayscaleDownscale(height=10, width=15, nb_shades=30)
go_explore = GoExplore(DQN, "MontezumaRevenge-v0", cell_factory, model_kwargs=dict(learning_starts=2000))
go_explore.explore(50000)

But you can expect major changes in the coming days on this repo.