Adding a few type annotations

automl · Oct 1, 2021 · d938c79 · d938c79
1 parent 644670c
commit d938c79
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/autoPyTorch/data/tabular_feature_validator.py b/autoPyTorch/data/tabular_feature_validator.py
@@ -22,9 +22,9 @@
 
 
 def _create_column_transformer(
-    preprocessors: Dict,
-    numerical_columns,
-    categorical_columns,
+    preprocessors: Dict[str],
+    numerical_columns: List[str],
+    categorical_columns: List[str],
 ) -> ColumnTransformer:
 
     numerical_pipeline = 'drop'