fix setup_tool version
@@ -52,7 +52,7 @@ CONDA_GPU = {
"numba": "numba>=0.38.1",
"pytorch": "pytorch>=1.0.0",
"tensorflow": "tensorflow-gpu==1.12.0",
- "cudatoolkit": "cudatoolkit==9.2"
+ "cudatoolkit": "cudatoolkit==9.2",
}
PIP_BASE = {
@@ -73,7 +73,7 @@ PIP_BASE = {
"ipywebrtc": "ipywebrtc==0.4.3",
"pre-commit": "pre-commit>=1.14.4",
"scikit-learn": "scikit-learn>=0.19.0,<=0.20.3",
- "setuptools_scm": "setuptools_scm==3.2.",
+ "setuptools_scm": "setuptools_scm==3.2.0",
"sklearn-crfsuite": "sklearn-crfsuite>=0.3.6",
"spacy": "spacy>=2.1.4",
"spacy-models": (
@@ -111,9 +111,7 @@ if __name__ == "__main__":
formatter_class=argparse.RawDescriptionHelpFormatter,
)
parser.add_argument("--name", help="specify name of conda environment")
- parser.add_argument(
- "--gpu", action="store_true", help="include packages for GPU support"
- )
+ parser.add_argument("--gpu", action="store_true", help="include packages for GPU support")
args = parser.parse_args()
# set name for environment and output yaml file
