arogozhnikov / hep_ml

Machine Learning for High Energy Physics.
https://arogozhnikov.github.io/hep_ml/
Other
176 stars 64 forks source link

Avoid sklearn.cross_validation deprecation warning #47

Closed alexpearce closed 7 years ago

alexpearce commented 7 years ago

Wraps the import of the (new) sklearn.model_selection with a try/except, falling back to the (old) sklearn.cross_validation module on failure. Users upgrading only hep_ml shouldn't see any difference, but users installing hep_ml from scratch will no longer see the deprecation warning.

This fixes #46.

arogozhnikov commented 7 years ago

Thank you. (this fix was in the bunch of work, which I didn't yet finish)