--- a +++ b/myosuite/tests/test_sb.py @@ -0,0 +1,14 @@ +from stable_baselines3 import PPO +import myosuite +from myosuite.utils import gym + +from myosuite import myosuite_myobase_suite +from myosuite import myosuite_myochal_suite +for env_name in myosuite_myochal_suite:#myosuite_myobase_suite+myosuite_myochal_suite: + print("\n\n"+env_name + " ---- Starting policy learning") + env = gym.make(env_name) + model = PPO("MlpPolicy", env, verbose=0) + print("========================================") + model.learn(total_timesteps=2) + +