change pkg names
@@ -30,8 +30,8 @@ requirements:
- sentencepiece
- seqeval
- setuptools
- - sklearn
- - torch >1.5,<1.8
+ - scikit-learn
+ - pytorch >1.5,<1.8
- tqdm
- transformers ==4.1.1
- werkzeug ==0.16.1
@@ -51,8 +51,8 @@ requirements:
- sentencepiece
- seqeval
- setuptools
- - sklearn
- - torch >1.5,<1.8
+ - scikit-learn
+ - pytorch >1.5,<1.8
- tqdm
- transformers ==4.1.1
- werkzeug ==0.16.1
