Skip to content

Commit 2b910d5

Browse files
committed
Skip umap-learn upstream tests for scikit-learn 1.8.*.
1 parent 20a08cf commit 2b910d5

1 file changed

Lines changed: 15 additions & 0 deletions

File tree

python/cuml/cuml_accel_tests/upstream/umap/run-tests.sh

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,21 @@ set -eu
1414

1515
UMAP_TAG="release-0.5.7"
1616

17+
# Skip tests for scikit-learn >= 1.8 -- umap-learn is not compatible with scikit-learn 1.8 yet
18+
python -c "
19+
import sys
20+
from packaging.version import Version
21+
import sklearn
22+
sys.exit(
23+
int(
24+
Version(sklearn.__version__) >= Version('1.8')
25+
)
26+
)
27+
" || {
28+
echo "Skipping umap tests for scikit-learn >= 1.8"
29+
exit 0
30+
}
31+
1732
THIS_DIRECTORY=$( cd "$(dirname "${BASH_SOURCE[0]}")" ; pwd -P )
1833
UMAP_REPO="${THIS_DIRECTORY}/umap-upstream"
1934

0 commit comments

Comments
 (0)