AnimalAI + Stable Baselines 3

A template repository for integrating the AnimalAI environment with Stable Baselines 3 agents.

This project was configured using Rye, but it should work with any tool supporting pyproject.toml.

Note: Check the version requirements, they are quite particular.
Note: Confirmed working with python == 3.9
Note: Very modern GPUs might not be compatible because of the older Torch version.

Usage

• Playing rye run play
• Training rye run train

Check the [project.scripts] section of pyproject.toml for the relevant configuration.

Resources

• Docs for Stable Baselines 3 ==v2.0.0.