Removing remaining check_cupy8

rapidsai · Mar 30, 2021 · f2a884b · f2a884b
1 parent b919c84
commit f2a884b
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git a/python/cuml/_thirdparty/sklearn/preprocessing/_discretization.py b/python/cuml/_thirdparty/sklearn/preprocessing/_discretization.py
@@ -27,7 +27,6 @@
 from ..utils.validation import FLOAT_DTYPES
 from ..utils.validation import _deprecate_positional_args
 from ....thirdparty_adapters import check_array
-from ....common.import_utils import check_cupy8
 from ....common.array_sparse import SparseCumlArray
 from ....common.array_descriptor import CumlArrayDescriptor
 from ....common.memory_utils import using_output_type
@@ -138,7 +137,6 @@ class KBinsDiscretizer(TransformerMixin,
     bin_edges_ = CumlArrayDescriptor()
     n_bins_ = CumlArrayDescriptor()
 
-    @check_cupy8()
     @_deprecate_positional_args
     def __init__(self, n_bins=5, *, encode='onehot', strategy='quantile'):
         self.n_bins = n_bins

diff --git a/python/cuml/_thirdparty/sklearn/preprocessing/_imputation.py b/python/cuml/_thirdparty/sklearn/preprocessing/_imputation.py
@@ -26,7 +26,6 @@
 from ..utils.validation import check_is_fitted
 from ..utils.validation import FLOAT_DTYPES
 from ..utils.validation import _deprecate_positional_args
-from ....common.import_utils import check_cupy8
 from ....common.array_sparse import SparseCumlArray
 from ....common.array_descriptor import CumlArrayDescriptor
 
@@ -229,7 +228,6 @@ class SimpleImputer(_BaseImputer,
 
     statistics_ = CumlArrayDescriptor()
 
-    @check_cupy8()
     @_deprecate_positional_args
     def __init__(self, *, missing_values=np.nan, strategy="mean",
                  fill_value=None, copy=True, add_indicator=False):