From d3a718b94e771fa154d52d5e6e71e02d5fd1f8b3 Mon Sep 17 00:00:00 2001
From: Antonin Raffin <antonin.raffin@dlr.de>
Date: Wed, 8 Jan 2020 11:26:57 +0100
Subject: [PATCH] Add extra dependency

---
 setup.py                             | 3 +++
 torchy_baselines/common/save_util.py | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 71242ee..e5e411f 100644
--- a/setup.py
+++ b/setup.py
@@ -23,6 +23,9 @@ setup(name='torchy_baselines',
             'sphinx',
             'sphinx-autobuild',
             'sphinx-rtd-theme'
+        ],
+        'render': [
+            'opencv-python'
         ]
       },
       description='Pytorch version of Stable Baselines, implementations of reinforcement learning algorithms.',
diff --git a/torchy_baselines/common/save_util.py b/torchy_baselines/common/save_util.py
index 30dd0b2..7b6a2be 100644
--- a/torchy_baselines/common/save_util.py
+++ b/torchy_baselines/common/save_util.py
@@ -131,4 +131,4 @@ def json_to_data(json_string, custom_objects=None):
         else:
             # Read as it is
             return_data[data_key] = data_item
-    return return_data
\ No newline at end of file
+    return return_data