From 6cd8fa704d79b4c7ccf650e1ea567a386a0b75b0 Mon Sep 17 00:00:00 2001
From: BDonnot <benjamin.donnot@gmail.com>
Date: Tue, 16 Jun 2020 12:24:54 +0200
Subject: [PATCH] adding extra package requirement for required

---
 setup.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index f14847c..7bd14fb 100644
--- a/setup.py
+++ b/setup.py
@@ -13,7 +13,14 @@
 
 pkgs = {
     "required": [
-        "grid2op[optional]>=0.9.1.post1"
+        "grid2op[optional]>=0.9.1.post1",
+        "tensorflow>=2.2.0",
+        "Keras>=2.3.1",
+        "torch>=1.4.0",
+        "statsmodels>=0.11.1",
+        "scikit-learn>=0.22.2.post1",
+        "gym>=0.17.1",
+        "scipy>=1.4.1",
     ],
     "extras": {
         "docs": [