[TASK] Add tests for matching the exact dtype expected by the Scikit-learn API to all estimators.

rapidsai / cuml

cuML - RAPIDS Machine Learning Library

https://docs.rapids.ai/api/cuml/stable/

Apache License 2.0

4.25k stars 534 forks source link

[TASK] Add tests for matching the exact dtype expected by the Scikit-learn API to all estimators. #5947

Open dantegd opened 4 months ago

dantegd commented 4 months ago

Currently we test accuracy of results in general, but there are cases like #5946 where we return correct values with the wrong dtype, breaking interoperability. This task encompasses adding a set of small and fast tests for explicit output dtype matching for all/relevant estimators.