adding extra package requirement for required

Grid2op · Jun 16, 2020 · 6cd8fa7 · 6cd8fa7
1 parent ba7b82d
commit 6cd8fa7
Showing 1 changed file with 8 additions and 1 deletion.
diff --git 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": [