from setuptools import find_packages, setup

if __name__ == "__main__":
    setup(
        name="dxgb-bench",
        version="0.0.0",
        description="Benchmark scripts for XGBoost.",
        scripts=["dxgb-bench", "dxgb-to-csv"],
        install_requires=[
            "numpy",
            "pandas",
            # "cudf",
            "dask",
            "distributed",
            # "dask-cuda",
            # "dask-cudf",
            "pynvml",
            "psutil",
            "typing_extensions",
        ],
        packages=find_packages(),
    )
