Merge pull request #32 from srivatsankrishnan/environment-lib-fix

fix for torch and qhoptim by adding them in the requirements.txt
srivatsankrishnan · Oct 8, 2023 · da8d71a · da8d71a
2 parents 1162f65 + 3d10fa3
commit da8d71a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/environment.yml b/environment.yml
@@ -11,12 +11,8 @@ dependencies:
   - requests
   - tqdm >= 4.37.0
   - keras
-  - scikit-learn
   - psutil
-  - torch
-  - xgboost
   - beautifulsoup4
-  - qhoptim
   - lxml
   - pytz
   - nodejs

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,9 @@
 tensorboard >= 2.3
 tensorflow >= 2.3
+qhoptim >= 1.1
+torch >= 1.6
+scikit-learn
+xgboost
 atoma
 scikit-optimize
 envlogger