Requirements fix

CarperAI · Jul 10, 2023 · b00b680 · b00b680
1 parent 258b34d
commit b00b680
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -13,6 +13,14 @@ We want to support users with many different compute profiles!
 # Install
 `pip install openelm`
 
+To use the sodarace environment, you must first `pip install swig`.
+
+Then:
+
+`pip install openelm[sodaracer]`
+
+See the pyproject.toml for further install options.
+
 # Features
 
 ### LLM integration with evolutionary algorithms

diff --git a/pyproject.toml b/pyproject.toml
@@ -86,4 +86,8 @@ triton = [
     "tritonclient[all]",
 ]
 notebook = ["ipython"]
-sodaracer = ["pygame"]
+sodaracer = [
+    "swig>=4.1.0",
+    "box2d-py==2.3.8",
+    "pygame"
+]
diff --git a/requirements.txt b/requirements.txt
@@ -4,8 +4,6 @@ numpy
 torch>=1.10
 transformers>=4.22.0
 tokenizers
-swig>=4.1.0
-box2d-py==2.3.8
 requests
 Flask
 scikit-learn