added pip requirements for rl control

neurolib-dev · May 23, 2024 · 899616c · 899616c
1 parent e02094a
commit 899616c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/requirements_rl_control.txt b/requirements_rl_control.txt
@@ -0,0 +1,2 @@
+gymnasium
+matplotlib
diff --git a/setup.py b/setup.py
@@ -9,6 +9,9 @@
 with open("requirements.txt") as f:
     requirements = f.read().splitlines()
 
+with open("requirements_rl_control.txt") as f:
+    requirements_rl_control = f.read().splitlines()
+
 setuptools.setup(
     name="neurolib",
     version="0.6.1",
@@ -27,5 +30,8 @@
     ],
     python_requires=">=3.6",
     install_requires=requirements,
+    extras_require={
+        'rl_control': requirements_rl_control
+    }
     include_package_data=True,
 )