diff --git a/python/cudf/setup.py b/python/cudf/setup.py index 5c2bff92648..4a5a0d2186f 100644 --- a/python/cudf/setup.py +++ b/python/cudf/setup.py @@ -30,16 +30,16 @@ import versioneer install_requires = [ - "numba>=0.53.1", - "Cython>=0.29,<0.30", + "cachetools", + "cuda-python>=11.5,<12.0", "fsspec>=0.6.0", + "numba>=0.53.1", "numpy", - "pandas>=1.0,<1.5.0dev0", - "typing_extensions", - "protobuf", "nvtx>=0.2.1", - "cachetools", "packaging", + "pandas>=1.0,<1.5.0dev0", + "protobuf", + "typing_extensions", ] extras_require = { @@ -259,6 +259,6 @@ def run(self): ), cmdclass=cmdclass, install_requires=install_requires, - zip_safe=False, extras_require=extras_require, + zip_safe=False, ) diff --git a/python/cudf_kafka/pyproject.toml b/python/cudf_kafka/pyproject.toml index 9855188ac6c..630efd5b9ec 100644 --- a/python/cudf_kafka/pyproject.toml +++ b/python/cudf_kafka/pyproject.toml @@ -5,10 +5,9 @@ requires = [ "wheel", "setuptools", - "Cython>=0.29,<0.30", + "cython>=0.29,<0.30", ] - [tool.black] line-length = 79 target-version = ["py36"] diff --git a/python/cudf_kafka/setup.py b/python/cudf_kafka/setup.py index 48009b566bb..6416bfb550d 100644 --- a/python/cudf_kafka/setup.py +++ b/python/cudf_kafka/setup.py @@ -13,6 +13,8 @@ install_requires = ["cudf", "cython"] +extras_require = {"test": ["pytest", "pytest-xdist"]} + cython_files = ["cudf_kafka/_lib/*.pyx"] CUDA_HOME = os.environ.get("CUDA_HOME", False) @@ -94,7 +96,6 @@ "Programming Language :: Python :: 3.9", ], # Include the separately-compiled shared library - setup_requires=["Cython>=0.29,<0.30"], ext_modules=cythonize( extensions, nthreads=nthreads, @@ -109,6 +110,6 @@ ), cmdclass=versioneer.get_cmdclass(), install_requires=install_requires, - extras_require={"test": ["pytest", "pytest-xdist"]}, + extras_require=extras_require, zip_safe=False, ) diff --git a/python/custreamz/setup.py b/python/custreamz/setup.py index 9f22b270a1b..37a45729921 100644 --- a/python/custreamz/setup.py +++ b/python/custreamz/setup.py @@ -6,6 +6,8 @@ install_requires = ["cudf_kafka", "cudf"] +extras_require = {"test": ["pytest", "pytest-xdist"]} + setup( name="custreamz", version=versioneer.get_version(), @@ -26,6 +28,6 @@ packages=find_packages(include=["custreamz", "custreamz.*"]), cmdclass=versioneer.get_cmdclass(), install_requires=install_requires, + extras_require=extras_require, zip_safe=False, - extras_require={"test": ["pytest", "pytest-xdist"]}, )