We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 20a08cf commit 2b910d5Copy full SHA for 2b910d5
1 file changed
python/cuml/cuml_accel_tests/upstream/umap/run-tests.sh
@@ -14,6 +14,21 @@ set -eu
14
15
UMAP_TAG="release-0.5.7"
16
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
+
32
THIS_DIRECTORY=$( cd "$(dirname "${BASH_SOURCE[0]}")" ; pwd -P )
33
UMAP_REPO="${THIS_DIRECTORY}/umap-upstream"
34
0 commit comments