[[source]]
name = "pypi"
url = "https://pypi.org/simple"
verify_ssl = true

[scripts]
genienlp = "python3 -m genienlp"

[dev-packages]
scikit-learn = "*"

[packages]
numpy = ">=1.14.5"
torch = "~=1.7.1"
tqdm = "~=4.0"
pyrouge = ">=0.1.3"
sacrebleu = "~=1.0"
tensorboardX = "==2.1.*"
requests = "~=2.22"
transformers = "==4.1.1"
sentence-transformers = "==0.4.0"
radam = {git = "https://github.com/LiyuanLucasLiu/RAdam"}
loss_dropper = {git = "https://github.com/ddkang/loss_dropper"}
matplotlib = '~=3.1'
seaborn = '~=0.9'
sentencepiece = '==0.1.*'

[requires]
python_version = "3.8"