Fix TfidfTransformer.

rapidsai · Feb 9, 2023 · a1b804d · a1b804d
1 parent 78bbe59
commit a1b804d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/cuml/feature_extraction/_tfidf.py b/python/cuml/feature_extraction/_tfidf.py
@@ -188,7 +188,7 @@ def fit(self, X) -> "TfidfTransformer":
 
         return self
 
-    @cuml.internals.api_base_return_any_skipall
+    @cuml.internals.api_base_return_array()
     def transform(self, X, copy=True):
         """Transform a count matrix to a tf or tf-idf representation
 
@@ -238,7 +238,7 @@ def transform(self, X, copy=True):
 
         return X
 
-    @cuml.internals.api_base_return_any_skipall
+    @cuml.internals.api_base_return_array()
     def fit_transform(self, X, copy=True):
         """
         Fit TfidfTransformer to X, then transform X.