Check `test__predict_logits__output_has_correct_properties_and_consistent_with_proba` and make it work for 2.6 model
Check
test__predict_logits__output_has_correct_properties_and_consistent_with_probaand make it work for 2.6 model