Skip to content

Commit

Permalink
Enable all tests for arm64 jobs (#2248)
Browse files Browse the repository at this point in the history
* Enable all pytests for arm jobs

* drop files

* Update test_wheel_pylibraft.sh

* Update test_wheel_raft_dask.sh
  • Loading branch information
galipremsagar authored Apr 10, 2024
1 parent 56aabe3 commit 139bfd9
Show file tree
Hide file tree
Showing 4 changed files with 4 additions and 174 deletions.
9 changes: 2 additions & 7 deletions ci/test_wheel_pylibraft.sh
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
#!/bin/bash
# Copyright (c) 2023, NVIDIA CORPORATION.
# Copyright (c) 2023-2024, NVIDIA CORPORATION.

set -euo pipefail

Expand All @@ -10,9 +10,4 @@ RAPIDS_PY_WHEEL_NAME="pylibraft_${RAPIDS_PY_CUDA_SUFFIX}" rapids-download-wheels
# echo to expand wildcard before adding `[extra]` requires for pip
python -m pip install $(echo ./dist/pylibraft*.whl)[test]

# Run smoke tests for aarch64 pull requests
if [[ "$(arch)" == "aarch64" && "${RAPIDS_BUILD_TYPE}" == "pull-request" ]]; then
python ./ci/wheel_smoke_test_pylibraft.py
else
python -m pytest ./python/pylibraft/pylibraft/test
fi
python -m pytest ./python/pylibraft/pylibraft/test
9 changes: 2 additions & 7 deletions ci/test_wheel_raft_dask.sh
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
#!/bin/bash
# Copyright (c) 2023, NVIDIA CORPORATION.
# Copyright (c) 2023-2024, NVIDIA CORPORATION.

set -euo pipefail

Expand All @@ -14,9 +14,4 @@ python -m pip install --no-deps ./local-pylibraft-dep/pylibraft*.whl
# echo to expand wildcard before adding `[extra]` requires for pip
python -m pip install $(echo ./dist/raft_dask*.whl)[test]

# Run smoke tests for aarch64 pull requests
if [[ "$(arch)" == "aarch64" && "${RAPIDS_BUILD_TYPE}" == "pull-request" ]]; then
python ./ci/wheel_smoke_test_raft_dask.py
else
python -m pytest ./python/raft-dask/raft_dask/test
fi
python -m pytest ./python/raft-dask/raft_dask/test
53 changes: 0 additions & 53 deletions ci/wheel_smoke_test_pylibraft.py

This file was deleted.

107 changes: 0 additions & 107 deletions ci/wheel_smoke_test_raft_dask.py

This file was deleted.

0 comments on commit 139bfd9

Please sign in to comment.