Skip to content

Commit 88b1a32

Browse files
committed
Specify requirements for installation
1 parent 71d4023 commit 88b1a32

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

setup.py

+9-3
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,18 @@
11
from setuptools import setup, find_packages
22

3-
VERSION = "0.0.3"
3+
VERSION = "0.0.4"
44

55
with open("README.md") as readme_file:
66
readme = readme_file.read()
77

8-
with open("requirements.txt") as requirements_file:
9-
install_requires = [line.strip() for line in requirements_file]
8+
install_requires = [
9+
"torch~=1.7.1",
10+
"tqdm~=4.58.0",
11+
"numpy~=1.20.1",
12+
"pytorch-lightning==1.1.7",
13+
"wandb~=0.10.20",
14+
"hydra-core~=1.0.6",
15+
]
1016

1117
setup_args = dict(
1218
name="code2seq",

0 commit comments

Comments
 (0)