Yair - added error handling for NA y_train labels 3b7934d verified KaiquanMah commited on Feb 15, 2025