From d655b2458055e56dce22f2be919999c0f3535f2b Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Thu, 4 Feb 2021 03:39:13 -0600
Subject: [PATCH] Added requirement in setup.py

---
 env.yaml | 19 -------------------
 setup.py | 13 +++++++++++++
 2 files changed, 13 insertions(+), 19 deletions(-)
 delete mode 100644 env.yaml

diff --git a/env.yaml b/env.yaml
deleted file mode 100644
index 735790b..0000000
--- a/env.yaml
+++ /dev/null
@@ -1,19 +0,0 @@
-name: predtuner
-channels:
-  - pytorch
-  - defaults
-dependencies:
-  - matplotlib=3.3.2
-  - networkx=2.5
-  - python=3.8.5
-  - pytorch=1.7.0
-  - torchvision=0.8.1
-  - tqdm=4.50.2
-  - pandas=1.1.3
-  - pip=20.2.4
-  - wheel=0.35.1
-  - jsonpickle=1.5
-  - pip:
-      - argparse
-      - opentuner==0.8.3  # Must be 0.8.3, they fixed an important bug
-      - sqlalchemy==1.3
diff --git a/setup.py b/setup.py
index 8d08aa7..8790283 100644
--- a/setup.py
+++ b/setup.py
@@ -13,4 +13,17 @@ setuptools.setup(
     long_description_content_type="text/markdown",
     url="https://github.com/Evan-Zhao/predictive-tuner",
     packages=["predtuner"],
+    install_requires=[
+        "matplotlib>=3.3",
+        "networkx>=2.5",
+        "torch==1.7.1",
+        "torchvision==0.8.2",
+        "tqdm>=4.50",
+        "pandas>=1.1",
+        "jsonpickle>=1.5",
+        "argparse>=1.4",
+        "wheel",  # Use wheel to build the following: 
+        "opentuner==0.8.3",  # Must be 0.8.3, they fixed an important bug
+        "sqlalchemy==1.3"
+    ]
 )
-- 
GitLab