[REF] Fix unit tests for ML library

Rubix library was complaining that: Classifiers require categorical labels, continuous given