Fix TypeError of customized kfold method which needs 'y'

[thinkall]

Why are these changes needed?

To fix TypeError: StratifiedKFold.split() missing 1 required positional argument: 'y' reported in #1303 and #1447

Related issue number

Closes #1303 , #1447

Checks

• I've used pre-commit to lint the changes in this PR (note the same in integrated in our CI checks).
• I've included any doc changes needed for https://microsoft.github.io/FLAML/. See https://microsoft.github.io/FLAML/docs/Contribute#documentation to build and test documentation locally.
• I've added tests (if relevant) corresponding to the changes introduced in this PR.
• I've made sure all auto checks have passed.