ignore unknown categories in OHE in unittest

Author: mfeurer

tests/test_flows/test_flow.py

@@ -209,7 +209,8 @@ def test_sklearn_to_upload_to_flow(self):
         y = iris.target
 
         # Test a more complicated flow
-        ohe = sklearn.preprocessing.OneHotEncoder(categorical_features=[1])
+        ohe = sklearn.preprocessing.OneHotEncoder(categorical_features=[1],
+                                                  handle_unknown='ignore')
         scaler = sklearn.preprocessing.StandardScaler(with_mean=False)
         pca = sklearn.decomposition.TruncatedSVD()
         fs = sklearn.feature_selection.SelectPercentile(