Improve extras_require names to facilitate their usage
Related to
@@ -8,8 +8,8 @@ setup(
"pyyaml"
],
extras_require={
- "TensorFlow": ["tensorflow==1.6.0"],
- "TensorFlow (with CUDA support)": ["tensorflow-gpu==1.6.0"]
+ "tensorflow": ["tensorflow==1.6.0"],
+ "tensorflow_gpu": ["tensorflow-gpu==1.6.0"]
},
tests_require=[
"nose2"
