search

RatInABox-Lab / RatInABox

A python package for modelling locomotion in complex environments and spatially/velocity selective cell activity.
MIT License
175 stars 31 forks source link

TaskEnvironment #49

Closed SynapticSage closed 1 year ago

SynapticSage commented 1 year ago

This PR introduces a TaskEnvironment that inherits ratinabox.Environment and pettingzoo.env environments. The main goal is to provide a streamlined interface for encoding and executing tasks and to improve the integration of RIB with reinforcement learning tools compatible with Gym APIs, like Stable Baselines3 and Rllib.

TomGeorge1234 commented 1 year ago

excellent work. very excited to give this a go! feel free to keep PR as/when you change or improve things

thanks again for the excellent work here and for making it public, I think this will benefit a lot of users.