RobertTLange / gymnax

RL Environments in JAX 🌍
Apache License 2.0
577 stars 54 forks source link

Is there a method of making custom environments and registering them? #75

Open kmcho2019 opened 2 months ago

kmcho2019 commented 2 months ago

Hello,

I have been enjoying your project so far but, as an inexperienced person with the codebase I have a few questions about it. Gymnasium supports making your custom environments and registering them so that gymnax.make() can be used on this new environment. Does gymnax offer a similar functionality?

When trying to add custom environments I found that I had to modify the package source code itself. I was wondering if there was an easier method something along the lines of the system offered by gymnasium.

My project needs to generate a new environment dynamically so modifying the source code was a bit of hassle for me. So I was wondering if there was a native implementation or a workaround that is possible.