Skip to content

Commit ecafd2b

Browse files
committed
add default param checks for CountVectorizerModel
1 parent a7755fd commit ecafd2b

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

python/pyspark/ml/tests.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2096,6 +2096,11 @@ def test_java_params(self):
20962096
# NOTE: disable check_params_exist until there is parity with Scala API
20972097
ParamTests.check_params(self, cls(), check_params_exist=False)
20982098

2099+
# Additional classes that need explicit construction
2100+
from pyspark.ml.feature import CountVectorizerModel
2101+
ParamTests.check_params(self, CountVectorizerModel.from_vocabulary(['a'], 'input'),
2102+
check_params_exist=False)
2103+
20992104

21002105
def _squared_distance(a, b):
21012106
if isinstance(a, Vector):

0 commit comments

Comments
 (0)