diff --git a/setup.py b/setup.py
index 89a3b57..f40a7da 100644
--- a/setup.py
+++ b/setup.py
@@ -34,7 +34,7 @@ setup(name='torchy_baselines',
       license="MIT",
       long_description="",
       long_description_content_type='text/markdown',
-      version="0.0.4",
+      version="0.0.5a",
       )
 
 # python setup.py sdist
diff --git a/torchy_baselines/common/base_class.py b/torchy_baselines/common/base_class.py
index a6b9a41..89b3e98 100644
--- a/torchy_baselines/common/base_class.py
+++ b/torchy_baselines/common/base_class.py
@@ -283,6 +283,12 @@ class BaseRLModel(object):
         raise NotImplementedError()
 
     def set_random_seed(self, seed=None):
+        """
+        Set the seed of the pseudo-random generators
+        (python, numpy, pytorch, gym, action_space)
+
+        :param seed: (int)
+        """
         if seed is None:
             return
         set_random_seed(seed, using_cuda=self.device == th.device('cuda'))