diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 30b1e59e1e..0c132a66c2 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -142,6 +142,7 @@ jobs: pip install $(python .ci/scripts/get_compatible_scipy_version.py ${{ matrix.SKLEARN_VERSION }}) pyyaml if [ "${{ steps.set-env.outputs.DPCFLAG }}" == "" ]; then pip install dpctl==${{ env.DPCTL_VERSION }} dpnp==${{ env.DPNP_VERSION }}; fi pip list + python -c "import dpnp" - name: Sklearnex testing run: | source venv/bin/activate @@ -464,10 +465,11 @@ jobs: with: name: sklearnex_build_${{ env.UXL_PYTHONVERSION }} - name: Install PyTorch - if: contains(matrix.FRAMEWORKS, 'pytorch') + if: contains(matrix.FRAMEWORKS, 'numpy') run: | - pip install torch --index-url https://download.pytorch.org/whl/xpu - python -c "import torch; _=[print(torch.xpu.get_device_name(i)) for i in range(torch.xpu.device_count())]" + pip install dpctl==${{ env.DPCTL_VERSION }} + # pip install torch --index-url https://download.pytorch.org/whl/xpu + # python -c "import torch; _=[print(torch.xpu.get_device_name(i)) for i in range(torch.xpu.device_count())]" - name: Install daal4py/sklearnex run: pip install *.whl - name: Sklearnex testing diff --git a/onedal/cluster/tests/test_kmeans.py b/onedal/cluster/tests/test_kmeans.py index bc85490687..ae3c966dc6 100644 --- a/onedal/cluster/tests/test_kmeans.py +++ b/onedal/cluster/tests/test_kmeans.py @@ -57,7 +57,7 @@ def gen_one(c): return (cs, vs, data) @pytest.mark.parametrize("queue", get_queues()) - @pytest.mark.parametrize("dtype", [np.float32, np.float64]) + @pytest.mark.parametrize("dtype", [np.float64]) @pytest.mark.parametrize("n_dim", [3, 4, 17, 24]) @pytest.mark.parametrize("n_cluster", [9, 11, 32]) @pytest.mark.parametrize("pipeline", ["implicit", "external", "internal"])