for set_ in (strat_train_set, strat_test_set): set_.drop("income_cat", axis=1, inplace=True) KeyError: "['income_cat'] not found in axis"