Update package requires_dist to remove extras that are not required (#1727)

* Remove test/dev/docs from extras requires

* Update requirements/gpu.txt to use pip package names

* Add extra index url for rapids packages to tox.ini

Author: oliverholworthy

requirements/gpu.txt

@@ -1,5 +1,3 @@
--r base.txt
-cudf>=21.12
-dask-cudf>=21.12
-dask-cuda>=21.12
-cupy>=7
+cudf-cu11
+dask-cudf-cu11
+cupy-cuda11x

setup.py

@@ -76,15 +76,10 @@ def read_requirements(req_path, filename):
         return packages
 
 
-requirements = {
-    "cpu": read_requirements("requirements", "base.txt"),
+install_requires = read_requirements("requirements", "base.txt")
+extras_require = {
     "gpu": read_requirements("requirements", "gpu.txt"),
 }
-dev_requirements = {
-    "dev": read_requirements("requirements", "dev.txt"),
-    "test": read_requirements("requirements", "test.txt"),
-    "docs": read_requirements("requirements", "docs.txt"),
-}
 
 with open("README.md", encoding="utf8") as readme_file:
     long_description = readme_file.read()
@@ -109,9 +104,6 @@ def read_requirements(req_path, filename):
     cmdclass=cmdclass,
     ext_modules=ext_modules,
     zip_safe=False,
-    install_requires=requirements["cpu"],
-    extras_require={
-        **requirements,
-        **dev_requirements,
-    },
+    install_requires=install_requires,
+    extras_require=extras_require,
 )

tox.ini

@@ -9,6 +9,8 @@ requires = pybind11
 commands =
     pip install --upgrade pip
     pip install .
+setenv =
+    PIP_EXTRA_INDEX_URL=https://pypi.ngc.nvidia.com
 
 [testenv:test-cpu]
 ; Runs in: Github Actions