Fix tests

automl · Jun 16, 2021 · 15064f5 · 15064f5
1 parent 2fa0913
commit 15064f5
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/...pipeline/components/preprocessing/tabular_preprocessing/feature_preprocessing/__init__.py b/...pipeline/components/preprocessing/tabular_preprocessing/feature_preprocessing/__init__.py
@@ -114,8 +114,7 @@ def get_hyperparameter_search_space(self,
                                                              default_value=default)
             else:
                 # Truncated SVD requires n_features > n_components
-                if len(dataset_properties['numerical_columns']) if isinstance(
-                        dataset_properties['numerical_columns'], List) else 0 == 1:
+                if len(numerical_columns) == 1:
                     del available_['TruncatedSVD']
                 preprocessor = CSH.CategoricalHyperparameter('__choice__',
                                                              list(available_.keys()),