[ci skip] Simplify quickstart example

DLR-RM · May 11, 2020 · b1794eb · b1794eb
1 parent b02afd6
commit b1794eb
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -116,7 +116,6 @@ Here is a quick example of how to train and run PPO on a cartpole environment:
 import gym
 
 from stable_baselines3 import PPO
-from stable_baselines3.ppo import MlpPolicy
 
 env = gym.make('CartPole-v1')
 

diff --git a/docs/guide/quickstart.rst b/docs/guide/quickstart.rst
@@ -13,11 +13,10 @@ Here is a quick example of how to train and run A2C on a CartPole environment:
   import gym
 
   from stable_baselines3 import A2C
-  from stable_baselines3.a2c import MlpPolicy
 
   env = gym.make('CartPole-v1')
 
-  model = A2C(MlpPolicy, env, verbose=1)
+  model = A2C('MlpPolicy', env, verbose=1)
   model.learn(total_timesteps=10000)
 
   obs = env.reset()

diff --git a/setup.py b/setup.py
@@ -40,11 +40,10 @@
 import gym
 
 from stable_baselines3 import PPO
-from stable_baselines3.ppo import MlpPolicy
 
 env = gym.make('CartPole-v1')
 
-model = PPO(MlpPolicy, env, verbose=1)
+model = PPO('MlpPolicy', env, verbose=1)
 model.learn(total_timesteps=10000)
 
 obs = env.reset()