From 623f8215719f3ca63ccd8ea6112d101c65859cca Mon Sep 17 00:00:00 2001
From: Antonin RAFFIN <antonin.raffin@ensta.org>
Date: Fri, 8 May 2020 12:14:33 +0200
Subject: [PATCH] Update examples

---
 docs/guide/examples.rst                          | 8 ++++----
 stable_baselines3/common/atari_wrappers.py       | 1 -
 stable_baselines3/common/vec_env/base_vec_env.py | 1 -
 3 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/docs/guide/examples.rst b/docs/guide/examples.rst
index 2760c14..9d642b4 100644
--- a/docs/guide/examples.rst
+++ b/docs/guide/examples.rst
@@ -12,10 +12,10 @@ notebooks:
 -  `Full Tutorial <https://github.com/araffin/rl-tutorial-jnrr19>`_
 -  `All Notebooks <https://github.com/Stable-Baselines-Team/rl-colab-notebooks/tree/sb3>`_
 -  `Getting Started`_
+-  `Training, Saving, Loading`_
 -  `RL Baselines zoo`_
 
 
-.. -  `Training, Saving, Loading`_
 .. -  `Multiprocessing`_
 .. -  `Monitor Training and Plotting`_
 .. -  `Atari Games`_
@@ -36,10 +36,10 @@ notebooks:
 Basic Usage: Training, Saving, Loading
 --------------------------------------
 
-In the following example, we will train, save and load a DQN model on the Lunar Lander environment.
+In the following example, we will train, save and load a A2C model on the Lunar Lander environment.
 
-.. .. image:: ../_static/img/colab-badge.svg
-..    :target: https://colab.research.google.com/github/Stable-Baselines-Team/rl-colab-notebooks/blob/sb3/saving_loading_dqn.ipynb
+.. image:: ../_static/img/colab-badge.svg
+   :target: https://colab.research.google.com/github/Stable-Baselines-Team/rl-colab-notebooks/blob/sb3/saving_loading_dqn.ipynb
 
 
 .. figure:: https://cdn-images-1.medium.com/max/960/1*f4VZPKOI0PYNWiwt0la0Rg.gif
diff --git a/stable_baselines3/common/atari_wrappers.py b/stable_baselines3/common/atari_wrappers.py
index 2b69254..ec409e2 100644
--- a/stable_baselines3/common/atari_wrappers.py
+++ b/stable_baselines3/common/atari_wrappers.py
@@ -28,7 +28,6 @@ class AtariWrapper(gym.Wrapper):
     * Scale observation: optional
 
     :param env: (gym.Env) gym environment
-        env (Env): environment
     :param noop_max: (int): max number of no-ops
     :param frame_skip: (int): the frequency at which the agent experiences the game.
     :param screen_size: (int): resize Atari frame
diff --git a/stable_baselines3/common/vec_env/base_vec_env.py b/stable_baselines3/common/vec_env/base_vec_env.py
index adf2ec0..e86bb55 100644
--- a/stable_baselines3/common/vec_env/base_vec_env.py
+++ b/stable_baselines3/common/vec_env/base_vec_env.py
@@ -9,7 +9,6 @@ import cloudpickle
 from stable_baselines3.common import logger
 
 
-
 def tile_images(img_nhwc: Sequence[np.ndarray]) -> np.ndarray:  # pragma: no cover
     """
     Tile N images into one big PxQ image