From bcabd144d85ad4c9407ea8616a42bb42af458247 Mon Sep 17 00:00:00 2001 From: Jason Wang Date: Fri, 20 May 2022 21:44:30 -0700 Subject: [PATCH] Fixing unit test --- tests/sparkml/test_pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/sparkml/test_pipeline.py b/tests/sparkml/test_pipeline.py index 2713b862..c13cfa59 100644 --- a/tests/sparkml/test_pipeline.py +++ b/tests/sparkml/test_pipeline.py @@ -125,7 +125,7 @@ def test_model_pipeline_2_stage(self): stages = [] for col in cols: stages.append(StringIndexer(inputCol=col, outputCol=col+'_index', handleInvalid='skip')) - stages.append(OneHotEncoder(inputCols=[col+'_index'], outputCols=[col+'_vec'])) + stages.append(OneHotEncoder(inputCols=[col+'_index'], outputCols=[col+'_vec'], dropLast=False)) pipeline = Pipeline(stages=stages)