Testing and using UpkieWheelsEnv gym env

[perrin-isir]

Hi,
when I run ./tools/bazelisk run //agents/ppo_balancer:train, I get:
ModuleNotFoundError: No module named 'stable_baselines3'

I could install it, but I thought using bazel should prevent from having to do so.

Other question: if I want to develop my own python code that would import UpkieWheelsEnv, how do I do?
Can I use bazel to avoid having to install all the dependencies (gin-config, vulp, ...) manually?

[stephane-caron]

Thanks for mentioning it, indeed I haven't put it in requirements.txt.

This is a good incentive to switch to the more recent pip_parse command from Bazel's rules_python, so that SB3 and its dependencies are only downloaded if needed (i.e. to run the PPO balancer, as other agents don't depend on it). I've started working on this in #12

[stephane-caron]

Edit: moved to #15.

[stephane-caron]

I could install it, but I thought using bazel should prevent from having to do so.

This should work now 👌

Spinning your follow-up question in #15.