cuml
cuml copied to clipboard
[BUG] Flaky NN test
Seen in multiple CI runs, there seems to be a very minor mismatch in a ANN test:
=========================== short test summary info ============================
FAILED test_nearest_neighbors.py::test_ivfpq_pred[8-512-4000-False-4-32-8] - assert <array_equal: [3. 0. 2. ... 4. 2. 1.] [3. 0. 2. ... 4. 2. 1.] unit_tol=0.0001 total_tol=0.0001 with_sign=True>
+ where <array_equal: [3. 0. 2. ... 4. 2. 1.] [3. 0. 2. ... 4. 2. 1.] unit_tol=0.0001 total_tol=0.0001 with_sign=True> = array_equal(array([3., 0., 2., ..., 4., 2., 1.]), array([3., 0., 2., ..., 4., 2., 1.], dtype=float32))
Test with such parameters is disabled for release 23.10, to be enabled in 23.12.