search

RobertTLange / gymnax

RL Environments in JAX 🌍
Apache License 2.0
613 stars 61 forks source link

Support passing environment to RolloutWrapper #71

Open cool-RR opened 5 months ago

cool-RR commented 5 months ago

@RobertTLange What do you think about this? I wanted to use RolloutWrapper on my own environment.