dependencies.yaml

# Dependency list for https://github.com/rapidsai/dependency-file-generator
files:
  all:
    output: [conda]
    matrix:
      cuda: ["11.8", "12.0"]
      arch: [x86_64]
    includes:
      - checks
      - common_build
      - cpp_build
      - cudatoolkit
      - docs
      - python_build_cythonize
      - python_run_cugraph
      - python_run_pylibcugraph
      - python_run_nx_cugraph
      - python_run_cugraph_dgl
      - python_run_cugraph_pyg
      - test_notebook
      - test_python_common
      - test_python_cugraph
      - test_python_pylibcugraph
      - test_python_nx_cugraph
  checks:
    output: none
    includes:
      - checks
      - py_version
  docs:
    output: none
    includes:
      - cudatoolkit
      - docs
      - py_version
  test_cpp:
    output: none
    includes:
      - cudatoolkit
      - test_cpp
  test_notebooks:
    output: none
    includes:
      - cudatoolkit
      - py_version
      - test_notebook
      - test_python_common
      - test_python_cugraph
  test_python:
    output: none
    includes:
      - cudatoolkit
      - py_version
      - test_python_common
      - test_python_cugraph
      - test_python_pylibcugraph
  py_build_cugraph:
    output: pyproject
    pyproject_dir: python/cugraph
    extras:
      table: build-system
    includes:
      - common_build
      - python_build_wheel
      - python_build_cythonize
      - python_build_cugraph
  py_run_cugraph:
    output: pyproject
    pyproject_dir: python/cugraph
    extras:
      table: project
    includes:
      - python_run_cugraph
  py_test_cugraph:
    output: pyproject
    pyproject_dir: python/cugraph
    extras:
      table: project.optional-dependencies
      key: test
    includes:
      - test_python_common
      - test_python_cugraph
  py_build_pylibcugraph:
    output: pyproject
    pyproject_dir: python/pylibcugraph
    extras:
      table: build-system
    includes:
      - common_build
      - python_build_wheel
      - python_build_cythonize
  py_run_pylibcugraph:
    output: pyproject
    pyproject_dir: python/pylibcugraph
    extras:
      table: project
    includes:
      - python_run_pylibcugraph
  py_test_pylibcugraph:
    output: pyproject
    pyproject_dir: python/pylibcugraph
    extras:
      table: project.optional-dependencies
      key: test
    includes:
      - test_python_common
      - test_python_pylibcugraph
  py_build_nx_cugraph:
    output: pyproject
    pyproject_dir: python/nx-cugraph
    extras:
      table: build-system
    includes:
      - python_build_wheel
  py_run_nx_cugraph:
    output: pyproject
    pyproject_dir: python/nx-cugraph
    extras:
      table: project
    includes:
      - python_run_nx_cugraph
  py_test_nx_cugraph:
    output: pyproject
    pyproject_dir: python/nx-cugraph
    extras:
      table: project.optional-dependencies
      key: test
    includes:
      - test_python_common
      - test_python_nx_cugraph
  py_build_cugraph_dgl:
    output: pyproject
    pyproject_dir: python/cugraph-dgl
    extras:
      table: build-system
    includes:
      - python_build_wheel
  py_run_cugraph_dgl:
    output: pyproject
    pyproject_dir: python/cugraph-dgl
    extras:
      table: project
    includes:
      - python_run_cugraph_dgl
  py_build_cugraph_pyg:
    output: pyproject
    pyproject_dir: python/cugraph-pyg
    extras:
      table: build-system
    includes:
      - python_build_wheel
  py_run_cugraph_pyg:
    output: pyproject
    pyproject_dir: python/cugraph-pyg
    extras:
      table: project
    includes:
      - python_run_cugraph_pyg
  py_build_cugraph_service_client:
    output: pyproject
    pyproject_dir: python/cugraph-service/client
    extras:
      table: build-system
    includes:
      - python_build_wheel
  py_run_cugraph_service_client:
    output: pyproject
    pyproject_dir: python/cugraph-service/client
    extras:
      table: project
    includes:
      - python_run_cugraph_service_client
  py_build_cugraph_service_server:
    output: pyproject
    pyproject_dir: python/cugraph-service/server
    extras:
      table: build-system
    includes:
      - python_build_wheel
  py_run_cugraph_service_server:
    output: pyproject
    pyproject_dir: python/cugraph-service/server
    extras:
      table: project
    includes:
      - python_run_cugraph_service_server
  py_test_cugraph_service_server:
    output: pyproject
    pyproject_dir: python/cugraph-service/server
    extras:
      table: project.optional-dependencies
      key: test
    includes:
      - test_python_common
      - test_python_cugraph
  cugraph_dgl_dev:
    matrix:
      cuda: ["11.8"]
    output: conda
    conda_dir: python/cugraph-dgl/conda
    includes:
      - checks
      - cugraph_dgl_dev
      - test_python_common
  cugraph_pyg_dev:
    matrix:
      cuda: ["11.8"]
    output: conda
    conda_dir: python/cugraph-pyg/conda
    includes:
      - checks
      - cugraph_pyg_dev
      - test_python_common
channels:
  - rapidsai
  - rapidsai-nightly
  - dask/label/dev
  - pytorch
  - pyg
  - dglteam/label/cu118
  - conda-forge
  - nvidia
dependencies:
  checks:
    common:
      - output_types: [conda, requirements]
        packages:
          - pre-commit
  cudatoolkit:
    specific:
      - output_types: [conda]
        matrices:
          - matrix:
              cuda: "12.0"
            packages:
              - cuda-version=12.0
          - matrix:
              cuda: "11.8"
            packages:
              - cuda-version=11.8
              - cudatoolkit
          - matrix:
              cuda: "11.5"
            packages:
              - cuda-version=11.5
              - cudatoolkit
          - matrix:
              cuda: "11.4"
            packages:
              - cuda-version=11.4
              - cudatoolkit
          - matrix:
              cuda: "11.2"
            packages:
              - cuda-version=11.2
              - cudatoolkit
  common_build:
    common:
      - output_types: [conda, pyproject]
        packages:
          - &cmake_ver cmake>=3.26.4
          - ninja
  cpp_build:
    common:
      - output_types: [conda]
        packages:
          - c-compiler
          - cxx-compiler
          - gmock>=1.13.0
          - gtest>=1.13.0
          - libcugraphops==23.10.*
          - libraft-headers==23.10.*
          - libraft==23.10.*
          - librmm==23.10.*
          - openmpi # Required for building cpp-mgtests (multi-GPU tests)
    specific:
      - output_types: [conda]
        matrices:
          - matrix:
              arch: x86_64
            packages:
              - gcc_linux-64=11.*
          - matrix:
              arch: aarch64
            packages:
              - gcc_linux-aarch64=11.*
      - output_types: [conda]
        matrices:
          - matrix:
              arch: x86_64
              cuda: "11.8"
            packages:
              - nvcc_linux-64=11.8
          - matrix:
              arch: aarch64
              cuda: "11.8"
            packages:
              - nvcc_linux-aarch64=11.8
          - matrix:
              cuda: "12.0"
            packages:
              - cuda-version=12.0
              - cuda-nvcc
  docs:
    common:
      - output_types: [conda]
        packages:
          - doxygen
          - graphviz
          - ipython
          - nbsphinx
          - numpydoc
          - pydata-sphinx-theme
          - recommonmark
          - sphinx-copybutton
          - sphinx-markdown-tables
          - sphinx<6
          - sphinxcontrib-websupport
          - pylibcugraphops==23.10.*
  py_version:
    specific:
      - output_types: [conda]
        matrices:
          - matrix:
              py: "3.9"
            packages:
              - python=3.9
          - matrix:
              py: "3.10"
            packages:
              - python=3.10
          - matrix:
            packages:
              - python>=3.9,<3.11
  python_build_wheel:
    common:
      - output_types: [conda, pyproject]
        packages:
          - wheel
          - setuptools>=61.0.0
  python_build_cythonize:
    common:
      - output_types: [conda, pyproject]
        packages:
          - cython>=3.0.0
          - &pylibraft pylibraft==23.10.*
          - &rmm rmm==23.10.*
          - scikit-build>=0.13.1
  python_build_cugraph:
    common:
      - output_types: [conda, pyproject]
        packages:
          - &pylibcugraph pylibcugraph==23.10.*
  python_run_cugraph:
    common:
      - output_types: [conda, pyproject]
        packages:
          - &cudf cudf==23.10.*
          - &dask dask>=2023.7.1
          - &distributed distributed>=2023.7.1
          - &dask_cuda dask-cuda==23.10.*
          - &dask_cudf dask-cudf==23.10.*
          - &numba numba>=0.57
          - raft-dask==23.10.*
          - *rmm
          - &ucx_py ucx-py==0.34.*
      - output_types: conda
        packages:
          - aiohttp
          - &cupy cupy>=12.0.0
          - &dask-core dask-core>=2023.7.1
          - fsspec>=0.6.0
          - libcudf==23.10.*
          - requests
          - nccl>=2.9.9
          - ucx-proc=*=gpu
      - output_types: pyproject
        packages:
          - &cupy_pip cupy-cuda11x>=12.0.0
            # cudf uses fsspec but is protocol independent. cugraph
            # dataset APIs require [http] extras for use with cudf.
          - fsspec[http]>=0.6.0
          - *pylibcugraph
  python_run_pylibcugraph:
    common:
      - output_types: [conda, pyproject]
        packages:
          - *pylibraft
          - *rmm
  python_run_nx_cugraph:
    common:
      - output_types: [conda, pyproject]
        packages:
          - networkx>=3.0
      - output_types: conda
        packages:
          - *cupy
      - output_types: pyproject
        packages:
          - *cupy_pip
          - *pylibcugraph
  python_run_cugraph_dgl:
    common:
      - output_types: [conda, pyproject]
        packages:
          - *numba
          - &numpy numpy>=1.21
      - output_types: [pyproject]
        packages:
          - &cugraph cugraph==23.10.*
  python_run_cugraph_pyg:
    common:
      - output_types: [conda, pyproject]
        packages:
          - *numba
          - *numpy
      - output_types: [pyproject]
        packages:
          - *cugraph
  python_run_cugraph_service_client:
    common:
      - output_types: [conda, pyproject]
        packages:
          - &thrift thriftpy2
  python_run_cugraph_service_server:
    common:
      - output_types: [conda, pyproject]
        packages:
          - *cudf
          - *dask
          - *dask_cuda
          - *dask_cudf
          - *distributed
          - *numba
          - *numpy
          - *rmm
          - *thrift
          - *ucx_py
      - output_types: conda
        packages:
          - *cupy
          - *dask-core
      - output_types: pyproject
        packages:
          - *cupy_pip
          - *cugraph
          - cugraph-service-client==23.10.*
  doc:
    common:
      - output_types: [conda]
        packages:
          - doxygen
          - nbsphinx
          - numpydoc
          - pydata-sphinx-theme
          - recommonmark
          - sphinx
          - sphinxcontrib-websupport
          - sphinx-markdown-tables
          - sphinx-copybutton
          - pylibcugraphops==23.10.*
  test_cpp:
    common:
      - output_types: conda
        packages:
          - *cmake_ver
  test_notebook:
    common:
      - output_types: [conda, requirements]
        packages:
          - ipython
          - notebook>=0.5.0
  test_python_common:
    common:
      - output_types: [conda, pyproject]
        packages:
          - pandas
          - pytest
          - pytest-benchmark
          - pytest-cov
          - pytest-xdist
          - scipy
  test_python_cugraph:
    common:
      - output_types: [conda, pyproject]
        packages:
          - networkx>=2.5.1
          - *numpy
          - python-louvain
          - scikit-learn>=0.23.1
  test_python_pylibcugraph:
    common:
      - output_types: [conda, pyproject]
        packages:
          - *cudf
          - *numpy
  test_python_nx_cugraph:
    common:
      - output_types: [conda, pyproject]
        packages:
          - packaging>=21
            # not needed by nx-cugraph tests, but is required for running networkx tests
          - pytest-mpl
  cugraph_dgl_dev:
    common:
      - output_types: [conda]
        packages:
          - cugraph==23.10.*
          - pylibcugraphops==23.10.*
          - pytorch>=2.0
          - pytorch-cuda==11.8
          - dgl>=1.1.0.cu*
  cugraph_pyg_dev:
    common:
      - output_types: [conda]
        packages:
          - cugraph==23.10.*
          - pylibcugraphops==23.10.*
          - pytorch==2.0
          - pytorch-cuda==11.8
          - pyg=2.3.1=*torch_2.0.0*cu118*