From ef59a7e431293a5904197f51578b34c4db4e38b3 Mon Sep 17 00:00:00 2001
From: Antonin Raffin <antonin.raffin@dlr.de>
Date: Mon, 18 Nov 2019 15:11:19 +0100
Subject: [PATCH] Update version + add docstring

---
 setup.py                              | 2 +-
 torchy_baselines/common/base_class.py | 6 ++++++
 2 files changed, 7 insertions(+), 1 deletion(-)

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'))