Robosuite Costum env_Env not found

[bhe1004]

Hi! Thanks for sharing your great project.

I was looking for projects to study reinforcement learning in robosuite environment and came across this project. I installed setup.py and requirements after git clone, but I got an error running main_model_learn.py.
When creating the env with suite.make, it says that the env 'PegInHoleSmall ' was not found.
Please let me know if I'm missing something and how I can add a custom env to robosuite.

Thank you.