From 3d10fa39abbbbf1e436448124be4ab02991ede51 Mon Sep 17 00:00:00 2001
From: a-saraf <a-saraf@github.com>
Date: Fri, 6 Oct 2023 01:47:03 +0530
Subject: [PATCH] fix for torch and qhoptim by adding them in the
 requirements.txt

---
 environment.yml  | 4 ----
 requirements.txt | 4 ++++
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/environment.yml b/environment.yml
index 4f72dedc..99904480 100644
--- 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
index 3e8057dc..b23c3151 100644
--- 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