Skip to content

Commit

Permalink
Update dask-cuda version and disable wheel builds in CI (#3009)
Browse files Browse the repository at this point in the history
We need to install dask-cuda from source during wheel builds for cugraph CI to pass because dask evolves too fast to pin. We also don't want to build wheels on every PR, so we'll need that disabled. I'll make that change once I confirm that wheel builds pass tests here.

This PR also fixes an unrelated issue causing doctests to fail. I'm guessing that because the docstring is in tests/mg it isn't being discovered during conda gpuCI because we exclude that directory from the pytest command in ci/test.sh.

Authors:
  - Vyas Ramasubramani (https://github.com/vyasr)

Approvers:
  - Brad Rees (https://github.com/BradReesWork)
  - Sevag H (https://github.com/sevagh)

URL: #3009
  • Loading branch information
vyasr authored Dec 1, 2022
1 parent cb0d092 commit 3a14275
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 8 deletions.
11 changes: 4 additions & 7 deletions .github/workflows/wheels.yml
Original file line number Diff line number Diff line change
@@ -1,9 +1,6 @@
name: cuGraph wheels

on:
push:
branches:
- 'pull-request/[0-9]+'
workflow_call:
inputs:
versioneer-override:
Expand Down Expand Up @@ -35,7 +32,7 @@ jobs:
with:
repo: rapidsai/cugraph

build-type: ${{ inputs.build-type || 'pull-request' }}
build-type: ${{ inputs.build-type }}
branch: ${{ inputs.branch }}
sha: ${{ inputs.sha }}
date: ${{ inputs.date }}
Expand Down Expand Up @@ -68,7 +65,7 @@ jobs:
with:
repo: rapidsai/cugraph

build-type: ${{ inputs.build-type || 'pull-request' }}
build-type: ${{ inputs.build-type }}
branch: ${{ inputs.branch }}
sha: ${{ inputs.sha }}
date: ${{ inputs.date }}
Expand All @@ -89,9 +86,9 @@ jobs:
skbuild-configure-options: "-DDETECT_CONDA_ENV=OFF -DCUGRAPH_BUILD_WHEELS=ON -DCPM_cugraph-ops_SOURCE=/project/cugraph-ops/"

# Always want to test against latest dask/distributed.
test-before-amd64: "pip install git+https://github.com/dask/dask.git@main git+https://github.com/dask/distributed.git@main"
test-before-amd64: "pip install git+https://github.com/dask/dask.git@main git+https://github.com/dask/distributed.git@main git+https://github.com/rapidsai/dask-cuda.git@branch-22.12"
# On arm also need to install cupy from the specific webpage.
test-before-arm64: "pip install cupy-cuda11x -f https://pip.cupy.dev/aarch64 && pip install git+https://github.com/dask/dask.git@main git+https://github.com/dask/distributed.git@main"
test-before-arm64: "pip install cupy-cuda11x -f https://pip.cupy.dev/aarch64 && pip install git+https://github.com/dask/dask.git@main git+https://github.com/dask/distributed.git@main git+https://github.com/rapidsai/dask-cuda.git@branch-22.12"
test-extras: test
test-unittest: "RAPIDS_DATASET_ROOT_DIR=`pwd`/datasets pytest -v ./python/cugraph/cugraph/tests"
secrets: inherit
2 changes: 1 addition & 1 deletion python/cugraph/cugraph/dask/cores/k_core.py
Original file line number Diff line number Diff line change
Expand Up @@ -112,7 +112,7 @@ def k_core(input_graph, k=None, core_number=None, degree_type="bidirectional"):
... chunksize=chunksize, delimiter=" ",
... names=["src", "dst", "value"],
... dtype=["int32", "int32", "float32"])
>>> dg = cugraph.Graph(directed=True)
>>> dg = cugraph.Graph(directed=False)
>>> dg.from_dask_cudf_edgelist(ddf, source='src', destination='dst',
... edge_attr='value')
>>> KCore_df = dcg.k_core(dg)
Expand Down

0 comments on commit 3a14275

Please sign in to comment.