Skip to content

Commit 10aea66

Browse files
committed
case when all nans in test
1 parent 3564fa1 commit 10aea66

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

autoPyTorch/data/tabular_feature_validator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -521,7 +521,7 @@ def _is_datasets_consistent(self, diff_cols: List[Union[int, str]], X: pd.DataFr
521521
if len(diff_cols) == 0:
522522
return True
523523
else:
524-
return False
524+
return all(X[diff_cols].isna().all())
525525

526526
# dtype is different ==> the column in at least either of train or test datasets must be all NaN
527527
# inconsistent <==> dtype is different and the col in both train and test is not all NaN

0 commit comments

Comments
 (0)