build wheel

rapidsai · Dec 14, 2023 · 834c1ba · 834c1ba
1 parent 3c58d7c
commit 834c1ba
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 3 deletions.
diff --git a/.github/workflows/pr.yaml b/.github/workflows/pr.yaml
@@ -27,6 +27,7 @@ jobs:
       - wheel-tests-nx-cugraph
       - devcontainer
       - conda-python-build-cugraph-equivariant
+      - wheel-build-cugraph-equivariant
     secrets: inherit
     uses: rapidsai/shared-workflows/.github/workflows/[email protected]
   checks:
@@ -134,6 +135,12 @@ jobs:
     with:
       build_type: pull-request
       script: ci/test_wheel_nx-cugraph.sh
+  wheel-build-cugraph-equivariant:
+    secrets: inherit
+    uses: rapidsai/shared-workflows/.github/workflows/[email protected]
+    with:
+      build_type: pull-request
+      script: ci/build_wheel_cugraph-equivariant.sh
   devcontainer:
     secrets: inherit
     uses: rapidsai/shared-workflows/.github/workflows/[email protected]

diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh
@@ -55,7 +55,7 @@ cd "${package_dir}"
 python -m pip wheel . -w dist -vvv --no-deps --disable-pip-version-check
 
 # pure-python packages should not have auditwheel run on them.
-if [[ ${package_name} == "nx-cugraph" ]]; then
+if [[ ${package_name} == "nx-cugraph" || ${package_name} == "cugraph-equivariant" ]]; then
     RAPIDS_PY_WHEEL_NAME="${package_name}_${RAPIDS_PY_CUDA_SUFFIX}" rapids-upload-wheels-to-s3 dist
 else
     mkdir -p final_dist

diff --git a/ci/build_wheel_cugraph-equivariant.sh b/ci/build_wheel_cugraph-equivariant.sh
@@ -0,0 +1,6 @@
+#!/bin/bash
+# Copyright (c) 2023, NVIDIA CORPORATION.
+
+set -euo pipefail
+
+./ci/build_wheel.sh cugraph-equivariant python/cugraph-equivariant
diff --git a/conda/recipes/cugraph-equivariant/meta.yaml b/conda/recipes/cugraph-equivariant/meta.yaml
@@ -22,9 +22,7 @@ requirements:
   host:
     - python
   run:
-    - e3nn >=0.5.1
     - pylibcugraphops ={{ minor_version }}
-    - pytorch >=2.0
     - python
 
 tests:
-Original file line number
+Diff line change
@@ Expand Up / @@ -22,9 +22,7 @@ requirements: @@
       host:
         - python
       run:
-        - e3nn >=0.5.1
         - pylibcugraphops ={{ minor_version }}
-        - pytorch >=2.0
         - python
     tests:
@@ Expand Down @@