Denys88 / rl_games

RL implementations
MIT License
848 stars 142 forks source link

Added masks for envs #195

Open Denys88 opened 2 years ago

Denys88 commented 2 years ago

It was requested in this ticket: https://github.com/Denys88/rl_games/issues/194 Two things should be done: 1) env_info returned from the vec_env should have 'env_masks' : True 2) method get_env_masks(self) should be implemented

Denys88 commented 2 years ago

cc @ankurhanda and @ArthurAllshire

ankurhanda commented 2 years ago

Would be good to have a unit test on this as this feature is important.