Fix base testing

rapidsai · Mar 26, 2021 · d199019 · d199019
1 parent 1c5f422
commit d199019
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/cuml/test/test_base.py b/python/cuml/test/test_base.py
@@ -20,7 +20,8 @@
 import numpydoc.docscrape
 from cuml.test.utils import (get_classes_from_package,
                              small_classification_dataset)
-from cuml._thirdparty.sklearn.utils.skl_dependencies import sklBaseEstimator
+from cuml._thirdparty.sklearn.utils.skl_dependencies import BaseEstimator \
+                                                            as sklBaseEstimator
 
 all_base_children = get_classes_from_package(cuml, import_sub_packages=True)