search

tkipf / c-swm

Contrastive Learning of Structured World Models
MIT License
386 stars 67 forks source link

Add envs to python path #1

Closed mwufi closed 4 years ago

mwufi commented 4 years ago

Hi Thomas! I was testing this code, and for me it said

ModuleNotFoundError: No module named 'envs'

Please advise if this looks like a good fix?

mwufi commented 4 years ago

Also, since the default directory is checkpoints/cubes, I made it not throw an error if checkpoints folder doesn't exist :)

tkipf commented 4 years ago

Fantastic, thanks!