Skip to content

GitLab

Commit 39bd5caf authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

make pycountry optional dep

parent adde09b3
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
setup.py
View file @ 39bd5caf
...@@ -15,7 +15,7 @@ extras_require = { ...@@ -15,7 +15,7 @@ extras_require = {
], ],
"testing": ["pytest", "pytest-cov", "pytest-xdist"], "testing": ["pytest", "pytest-cov", "pytest-xdist"],
"multilingual": ["nagisa>=0.2.7", "jieba>=0.42.1"], "multilingual": ["nagisa>=0.2.7", "jieba>=0.42.1"],
"sentencepiece": ["sentencepiece>=0.1.98", "protobuf>=4.22.1"], "sentencepiece": ["sentencepiece>=0.1.98", "protobuf>=4.22.1", "pycountry"],
"promptsource": [ "promptsource": [
"promptsource @ git+https://github.com/bigscience-workshop/promptsource.git#egg=promptsource" "promptsource @ git+https://github.com/bigscience-workshop/promptsource.git#egg=promptsource"
], ],
...@@ -62,10 +62,9 @@ setuptools.setup( ...@@ -62,10 +62,9 @@ setuptools.setup(
"omegaconf>=2.2", "omegaconf>=2.2",
"peft>=0.2.0", "peft>=0.2.0",
"pybind11>=2.6.2", "pybind11>=2.6.2",
"pycountry",
"pytablewriter", "pytablewriter",
"rouge-score>=0.0.4", "rouge-score>=0.0.4",
"sacrebleu==1.5.0", "sacrebleu>=1.5.0",
"scikit-learn>=0.24.1", "scikit-learn>=0.24.1",
"sqlitedict", "sqlitedict",
"torch>=1.8", "torch>=1.8",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment