Skip to content

Commit 7e40b0a

Browse files
committed
[fix] Fix a mypy error
1 parent d8437e2 commit 7e40b0a

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

autoPyTorch/data/tabular_feature_validator.py

+1
Original file line numberDiff line numberDiff line change
@@ -143,6 +143,7 @@ def _encode_categories(self, X: pd.DataFrame) -> None:
143143
self.column_transformer.fit(X)
144144

145145
# The column transformer moves categoricals to the left side
146+
assert self.feat_type is not None # mypy definition
146147
self.feat_type = sorted(self.feat_type, key=functools.cmp_to_key(self._comparator))
147148

148149
encoded_categories = self.column_transformer.\

0 commit comments

Comments
 (0)