Skip to content

Commit

Permalink
more aggressive pin to packages from this CI
Browse files Browse the repository at this point in the history
  • Loading branch information
jameslamb committed Nov 4, 2024
1 parent 7494635 commit d08b924
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 12 deletions.
5 changes: 4 additions & 1 deletion ci/test_notebooks.sh
Original file line number Diff line number Diff line change
Expand Up @@ -26,11 +26,14 @@ rapids-logger "Downloading artifacts from previous jobs"
CPP_CHANNEL=$(rapids-download-conda-from-s3 cpp)
PYTHON_CHANNEL=$(rapids-download-conda-from-s3 python)

# TODO: remove the '>=24.12.00a1000' once we start publishing nightly packages
# from the 'cugraph-gnn' repo and stop publishing them from
# the 'cugraph' / 'wholegraph' repos
rapids-mamba-retry install \
--channel "${CPP_CHANNEL}" \
--channel "${PYTHON_CHANNEL}" \
--channel dglteam/label/th23_cu118 \
"cugraph-dgl=${RAPIDS_VERSION}"
"cugraph-dgl=${RAPIDS_VERSION},>=24.12.00a1000"

NBTEST="$(realpath "$(dirname "$0")/utils/nbtest.sh")"
NOTEBOOK_LIST="$(realpath "$(dirname "$0")/notebook_list.py")"
Expand Down
25 changes: 14 additions & 11 deletions ci/test_python.sh
Original file line number Diff line number Diff line change
Expand Up @@ -57,17 +57,18 @@ if [[ "${RUNNER_ARCH}" != "ARM64" ]]; then
conda activate test_cugraph_dgl
set -u

# TODO: remove the '>=24.12.00a1000' once we start publishing nightly packages
# from the 'cugraph-gnn' repo and stop publishing them from
# the 'cugraph' / 'wholegraph' repos
rapids-mamba-retry install \
--channel "${CPP_CHANNEL}" \
--channel "${PYTHON_CHANNEL}" \
--channel pytorch \
--channel conda-forge \
--channel dglteam/label/th23_cu118 \
--channel nvidia \
"pylibwholegraph=${RAPIDS_VERSION}" \
"pylibcugraphops=${RAPIDS_VERSION}" \
"cugraph=${RAPIDS_VERSION}" \
"cugraph-dgl=${RAPIDS_VERSION}" \
"pylibwholegraph=${RAPIDS_VERSION},>=24.12.00a1000" \
"cugraph-dgl=${RAPIDS_VERSION},>=24.12.00a1000" \
'pytorch::pytorch>=2.3,<2.4' \
"ogb"

Expand Down Expand Up @@ -100,15 +101,15 @@ if [[ "${RUNNER_ARCH}" != "ARM64" ]]; then
conda activate test_cugraph_pyg
set -u

# Will automatically install built dependencies of cuGraph-PyG
# TODO: remove the '>=24.12.00a1000' once we start publishing nightly packages
# from the 'cugraph-gnn' repo and stop publishing them from
# the 'cugraph' / 'wholegraph' repos
rapids-mamba-retry install \
--channel "${CPP_CHANNEL}" \
--channel "${PYTHON_CHANNEL}" \
--channel pytorch \
"pylibwholegraph=${RAPIDS_VERSION}" \
"pylibcugraphops=${RAPIDS_VERSION}" \
"cugraph=${RAPIDS_VERSION}" \
"cugraph-pyg=${RAPIDS_VERSION}" \
"pylibwholegraph=${RAPIDS_VERSION},>=24.12.00a1000" \
"cugraph-pyg=${RAPIDS_VERSION},>=24.12.00a1000" \
'pytorch::pytorch>=2.3,<2.4' \
'ogb'

Expand Down Expand Up @@ -141,13 +142,15 @@ if [[ "${RUNNER_ARCH}" != "ARM64" ]]; then
conda activate test_pylibwholegraph
set -u

# Will automatically install built dependencies of pylibwholegraph
# TODO: remove the '>=24.12.00a1000' once we start publishing nightly packages
# from the 'cugraph-gnn' repo and stop publishing them from
# the 'cugraph' / 'wholegraph' repos
rapids-mamba-retry install \
--channel "${CPP_CHANNEL}" \
--channel "${PYTHON_CHANNEL}" \
--channel pytorch \
'mkl<2024.1.0' \
"pylibwholegraph=${RAPIDS_VERSION}" \
"pylibwholegraph=${RAPIDS_VERSION},>=24.12.00a1000" \
'pytorch::pytorch>=2.3,<2.4' \
'pytest-forked' \
'ogb'
Expand Down

0 comments on commit d08b924

Please sign in to comment.